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.cc61 for (int64 dimension_number : minor_to_major) { in MakeLayout() local
Dshape_util.cc512 int64 dimension_number) { in GetDimension()
517 int64 dimension_number) { in GetDimensionNumber()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc379 for (int64 dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc215 int64 dimension_number = instr.dimensions(0); in IsConstantVisitor() local