Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc647 for (const auto& node_shape_info : shape_info) { in GetGraph() local
648 const string& node_name = node_shape_info.first; in GetGraph()
649 const std::vector<InferredShape>& output_shapes = node_shape_info.second; in GetGraph()
670 for (const auto& node_shape_info : shape_info) { in GetGraph() local
671 const string& node_name = node_shape_info.first; in GetGraph()
672 const std::vector<InferredShape>& output_shapes = node_shape_info.second; in GetGraph()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc71 for (const auto& node_shape_info : graph_shape_info) { in ConvertGraphShapeInfoToShapeMap() local
72 const string& node_name = node_shape_info.first; in ConvertGraphShapeInfoToShapeMap()
73 const std::vector<InferredShape>& output_shapes = node_shape_info.second; in ConvertGraphShapeInfoToShapeMap()