Searched refs:scatter_dims_to_operand_dims (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 199 << absl::StrJoin(dim_numbers.scatter_dims_to_operand_dims(), ",") in XlaScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 154 FindIndex(dim_numbers.scatter_dims_to_operand_dims(), i); in ExpandIndexVectorIntoOperandSpace()
|
| D | hlo_instructions.cc | 2415 string scatter_dims_to_operand_dims = StrCat( in ScatterDimensionNumbersToString() local
2417 StrJoin(scatter_dimension_numbers().scatter_dims_to_operand_dims(), ","), in ScatterDimensionNumbersToString()
2423 {update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims, in ScatterDimensionNumbersToString()
2432 absl::Span<const int64> scatter_dims_to_operand_dims, in MakeScatterDimNumbers() argument
2441 for (int64 scatter_dim_to_operand_dim : scatter_dims_to_operand_dims) { in MakeScatterDimNumbers()
|
| D | shape_inference.cc | 3137 dim_numbers.scatter_dims_to_operand_dims(i); in ValidateScatterDimensionNumbers()
3147 dim_numbers.scatter_dims_to_operand_dims().begin(), in ValidateScatterDimensionNumbers()
3148 dim_numbers.scatter_dims_to_operand_dims().end()); in ValidateScatterDimensionNumbers()
3155 StrJoin(dim_numbers.scatter_dims_to_operand_dims(), ", ")); in ValidateScatterDimensionNumbers()
|
| D | hlo_parser.cc | 1655 optional<std::vector<int64>> scatter_dims_to_operand_dims; in ParseInstructionRhs() local
1658 &scatter_dims_to_operand_dims}; in ParseInstructionRhs()
1676 /*scatter_dims_to_operand_dims=*/*scatter_dims_to_operand_dims, in ParseInstructionRhs()
|
| D | hlo_instructions.h | 1419 absl::Span<const int64> scatter_dims_to_operand_dims,
|
| D | hlo_parser_test.cc | 872 …es), update_window_dims={4,5,6,7,8}, inserted_window_dims={}, scatter_dims_to_operand_dims={0,1,2,… in CreateTestCases()
|
| D | hlo_evaluator_typed_visitor.h | 2116 FindIndex(dim_numbers_.scatter_dims_to_operand_dims(), i); in UpdateScatterIndexToInputIndex()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 475 .scatter_dims_to_operand_dims(); in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2115 …on, index_vector_dim, update_window_dims, inserted_window_dims, scatter_dims_to_operand_dims)` </b>
2126 …scatter_dims_to_operand_dims` | `ArraySlice<int64>` | A dimensions map from the scatter indices to…
2162 - `scatter_dims_to_operand_dims.size` must be equal to
2174 `S` using the `scatter_dims_to_operand_dims` map. More formally:
2175 1. `S`<sub>`in`</sub>[`scatter_dims_to_operand_dims`[`k`]] = `S`[`k`] if
2176 `k` < `scatter_dims_to_operand_dims.size`.
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1969 self.scatter_dims_to_operand_dims = []
|
| D | xla_client_test.py | 1686 dnums.scatter_dims_to_operand_dims.append(0)
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 475 repeated int64 scatter_dims_to_operand_dims = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1147 int64 operand_dim = dim_numbers.scatter_dims_to_operand_dims(i); in EmitScatter()
|