tenzr::txpr::AttributePayload¶

std::vector< int64_t > axes;

std::vector< int64_t > permutation;

std::vector< Slice > slices;

std::vector< int64_t > edge_padding_low;

std::vector< int64_t > edge_padding_high;

std::vector< int64_t > interior_padding;

int64_t scatter_axis {0};

int64_t gather_axis {0};

std::string scatter_update {"add"};

int64_t sort_dimension {0};

bool sort_descending {false};

bool sort_return_values {false};

int64_t concat_axis {0};

DType target_dtype {DType::kFloat32};

bool keepdims {false};

std::string custom_op_name;

double scalar_value {0.0};

std::vector< double > tensor_data;

std::vector< int64_t > int_tensor_data;

int64_t iota_dimension {-1};

std::vector< int64_t > window_dimensions;

std::vector< int64_t > window_strides;

std::vector< int64_t > base_dilations;

std::vector< int64_t > window_dilations;

std::string reduce_window_op {"add"};

std::vector< int64_t > lhs_dilation;

std::vector< int64_t > rhs_dilation;

std::vector< int64_t > conv_lhs_spec;

std::vector< int64_t > conv_rhs_spec;

std::vector< int64_t > conv_out_spec;

std::vector< int64_t > slice_sizes;

std::string collective_op {"sum"};

int64_t collective_dim {0};

int64_t split_dim {0};

int64_t concat_dim {0};

int64_t split_count {1};

std::vector< std::vector< int64_t > > replica_groups;