Home
last modified time | relevance | path

Searched defs:shape_indices (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dallocation_tracker.cc70 std::vector<ShapeIndex> shape_indices; in RegisterInternal() local
103 std::vector<ShapeIndex> shape_indices; in Unregister() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_padding_mapper.cc170 auto shape_indices = builder.getNamedAttr( in AnnotateFunctionArgumentsWithPaddings() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1679 auto shape_indices = in AddDynamicParameterBindings() local