Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.h154 static Status BuildTensorShapeMapFromGraph(const Graph& graph,
Dremote_fused_graph_execute_utils.cc489 /* static */ Status RemoteFusedGraphExecuteUtils::BuildTensorShapeMapFromGraph( in BuildTensorShapeMapFromGraph() function in tensorflow::RemoteFusedGraphExecuteUtils
595 BuildTensorShapeMapFromGraph(graph, shape_refiner, &tensor_shape_map)); in BuildAndAddTensorShapes()
Dremote_fused_graph_execute_utils_test.cc353 ASSERT_TRUE(RemoteFusedGraphExecuteUtils::BuildTensorShapeMapFromGraph( in TEST()