Home
last modified time | relevance | path

Searched defs:dimension_number (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.cc63 for (int64 dimension_number : minor_to_major) { in MakeLayout() local
Dshape_util.cc684 int64 dimension_number) { in GetDimension()
689 int64 dimension_number) { in GetDimensionNumber()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc379 const DotDimensionNumbers& dimension_number, in DotGeneralInternal()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc413 for (int64 dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local