Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util.cc62 for (int64_t dimension_number : minor_to_major) { in MakeLayout() local
H A Dshape_util.cc765 int64_t dimension_number) { in GetDimension()
770 int64_t dimension_number) { in GetDimensionNumber()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc500 const DotDimensionNumbers& dimension_number, in DotGeneralInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc394 for (int64_t dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local