Searched defs:offset_dims (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 50 std::vector<int64_t> offset_dims(dnums.offset_dims().begin(), in ConvertGatherDimensionNumbers() local
|
| D | hlo_utils.cc | 270 auto offset_dims = CreateDenseIntElementsAttrFromVector( in CreateGatherDimensionNumbers() local
|
| D | mlir_hlo_to_hlo.cc | 321 auto offset_dims = ConvertDenseIntAttr(input.offset_dims()); in Convert_dimension_numbers() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 254 HloInstruction* accumulator, absl::Span<const int64> offset_dims, in PermuteBatchAndOffsetDims()
|
| D | hlo_sharding_util.cc | 1002 std::vector<int64> offset_dims(dnums.offset_dims().begin(), in GatherOutputShardingFromDataOperand() local
1016 std::vector<int64> offset_dims(dnums.offset_dims().begin(), in GatherDataOperandShardingFromOutput() local
|
| D | hlo_instructions.cc | 2902 string offset_dims = in GatherDimensionNumbersToString() local
2920 absl::Span<const int64> offset_dims, in MakeGatherDimNumbers()
|
| D | hlo_parser.cc | 2597 optional<std::vector<int64>> offset_dims; in ParseInstructionRhs() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 277 SmallVector<int64_t, 4> offset_dims; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 493 repeated int64 offset_dims = 1; field
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1679 auto offset_dims = gather_op.dimension_numbers().offset_dims(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1197 SmallVector<int64_t, 4> offset_dims; in matchAndRewrite() local
1717 SmallVector<int64_t, 4> offset_dims; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 403 SmallVector<int64_t, 4> offset_dims( in GatherShapeInferImpl() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3399 std::vector<int64> offset_dims = {x_dims[is_nhwc ? 3 : 1]}; in TEST_F() local
|