Home
last modified time | relevance | path

Searched refs:shape_it (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc267 for (auto shape_it = shape_tree.begin(); shape_it != shape_tree.end(); in ToString() local
268 ++shape_it) { in ToString()
269 llvm::Value* val = shape_it->second; in ToString()
270 StrAppend(&s, " ", shape_it->first.ToString(), " -> ", in ToString()
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding.cc239 auto shape_it = shape_map->find(n->name()); in IsConstantFoldable() local
240 if (shape_it != shape_map->end()) { in IsConstantFoldable()
241 for (int64 i = 0; i < shape_it->second.size(); ++i) { in IsConstantFoldable()
242 const auto& out_shape = shape_it->second[i]; in IsConstantFoldable()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1086 for (auto shape_it : llvm::enumerate(shape_value)) { in Verify() local
1087 int64_t i = shape_it.index(); in Verify()
1088 auto value = shape_it.value().getSExtValue(); in Verify()
1102 for (auto shape_it : llvm::enumerate(shape_value)) { in Verify() local
1103 int i = shape_it.index(); in Verify()
1104 auto value = shape_it.value().getSExtValue(); in Verify()