Searched refs:dimension_number (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 684 int64 dimension_number) { in GetDimension() argument
685 return shape.dimensions(GetDimensionNumber(shape, dimension_number)); in GetDimension()
689 int64 dimension_number) { in GetDimensionNumber() argument
690 if (dimension_number < 0) { in GetDimensionNumber()
691 dimension_number += shape.rank(); in GetDimensionNumber()
693 CHECK_GE(dimension_number, 0); in GetDimensionNumber()
694 return dimension_number; in GetDimensionNumber()
|
| D | layout_util.cc | 63 for (int64 dimension_number : minor_to_major) { in MakeLayout() local
64 layout.add_minor_to_major(dimension_number); in MakeLayout()
|
| D | shape_util.h | 369 static int64 GetDimension(const Shape& shape, int64 dimension_number);
379 static int64 GetDimensionNumber(const Shape& shape, int64 dimension_number);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.h | 205 const DotDimensionNumbers& dimension_number,
|
| D | mlir_hlo_builder.cc | 379 const DotDimensionNumbers& dimension_number, in DotGeneralInternal() argument
385 ConvertDotDimensionNumbers(dimension_number, &builder_), in DotGeneralInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 413 for (int64 dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local
414 ++dimension_number) { in InferConcatOpShape()
415 if (arg_shape->dimensions(dimension_number) != in InferConcatOpShape()
416 shape->dimensions(dimension_number)) { in InferConcatOpShape()
417 if (dimension_number == dimension) { in InferConcatOpShape()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1123 const DotDimensionNumbers& dimension_number,
1128 const DotDimensionNumbers& dimension_number,
|