Home
last modified time | relevance | path

Searched refs:current_shape_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.h195 xla::ShapeIndex* current_shape_index,
210 const xla::ShapeIndex& current_shape_index);
Dmhlo_to_lhlo_with_xla.cc1400 xla::ShapeIndex* current_shape_index, SmallVectorImpl<Value>* values) { in GetOrCreateViewImpl() argument
1403 current_shape_index->push_back(i); in GetOrCreateViewImpl()
1405 instr, current_shape.tuple_shapes(i), current_shape_index, values)); in GetOrCreateViewImpl()
1406 current_shape_index->pop_back(); in GetOrCreateViewImpl()
1412 *current_shape_index)); in GetOrCreateViewImpl()
1417 instr->ToString(), current_shape_index->ToString()); in GetOrCreateViewImpl()