Searched refs:input_node_info (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transfer_utils.cc | 117 for (const std::pair<string, Tensor>& input_node_info : inputs) { in BuildFusedGraphDef() local
118 const Scope& scope = root.WithOpName(input_node_info.first); in BuildFusedGraphDef()
122 .Attr("dtype", input_node_info.second.dtype()) in BuildFusedGraphDef()
123 .Attr("shape", input_node_info.second.shape()); in BuildFusedGraphDef()
128 input_types.push_back(input_node_info.second.dtype()); in BuildFusedGraphDef()
|
| D | graph_transferer.cc | 151 for (const std::pair<string, Tensor>& input_node_info : in LoadGraphFromProto() local
155 graph_input_node_info.set_name(input_node_info.first); in LoadGraphFromProto()
156 graph_input_node_info.set_dtype(input_node_info.second.dtype()); in LoadGraphFromProto()
157 for (const int64 dim : ToTensorShapeArray(input_node_info.second.shape())) { in LoadGraphFromProto()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils_test.cc | 229 std::pair<string, Tensor> input_node_info; in TEST() local
230 input_node_info.first = NAME_A; in TEST()
231 input_node_info.second = Tensor(DT_FLOAT, {}); in TEST()
232 input_node_info.second.scalar<float>()() = 1.0f; in TEST()
233 const std::vector<std::pair<string, Tensor>> inputs{input_node_info}; in TEST()
250 std::pair<string, Tensor> input_node_info; in TEST() local
251 input_node_info.first = NAME_A; in TEST()
252 input_node_info.second = Tensor(DT_FLOAT, {}); in TEST()
253 const std::vector<std::pair<string, Tensor>> inputs{input_node_info}; in TEST()
|
| D | remote_fused_graph_execute_utils.cc | 304 for (const std::pair<string, Tensor>& input_node_info : in DryRunInferenceForAllNode() local
306 output_tensors.push_back(input_node_info.second); in DryRunInferenceForAllNode()
453 for (const std::pair<string, Tensor>& input_node_info : in PropagateShapeInference() local
455 if (node->name() == input_node_info.first) { in PropagateShapeInference()
460 input_node_info.second.shape(), &handle); in PropagateShapeInference()
|