Home
last modified time | relevance | path

Searched defs:refiner (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor_test.cc73 ShapeRefiner* refiner = nullptr, in ExpectError()
97 ShapeRefiner* refiner = nullptr, in ExpectOK()
1171 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1211 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1274 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1339 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1371 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1392 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1414 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1439 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
[all …]
Dgraph_constructor.cc170 GraphConstructor(const Options& opts, Graph* g, ShapeRefiner* refiner, in GraphConstructor()
403 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in NodeDefCopyingGraphConstructor()
431 ShapeRefiner* refiner, std::vector<std::pair<Node*, int>>* return_tensors, in NodeDefMovingGraphConstructor()
462 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in Construct()
480 const Options& opts, GraphDef&& graph_def, Graph* g, ShapeRefiner* refiner, in Construct()
1455 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph() local
1464 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph() local
1473 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph() local
1487 Graph* g, ShapeRefiner* refiner, in ImportGraphDef()
/external/tensorflow/tensorflow/cc/framework/
Dscope.cc45 ShapeRefiner* refiner, bool disable_shape_inference) in Impl()
57 const std::shared_ptr<ShapeRefiner>& refiner) in Impl()
68 ShapeRefiner* refiner = in NewRootScope() local
76 ShapeRefiner* refiner = in DisabledShapeInferenceScope() local
509 static Scope NewScope(Graph* graph, Status* status, ShapeRefiner* refiner) { in NewScope()
530 Scope NewInternalScope(Graph* graph, Status* status, ShapeRefiner* refiner) { in NewInternalScope()
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc39 const ShapeRefiner& refiner, in TryToInferTensorOutputFromInputShapes()
130 const Node& target_node, const ShapeRefiner& refiner, in ExtractConstantSubgraph()
311 Status EvaluateConstantTensor(OutputTensor tensor, const ShapeRefiner& refiner, in EvaluateConstantTensor()
Dshape_refiner.cc66 Status InferShapesForFunctionSubNode(const Node* node, ShapeRefiner* refiner, in InferShapesForFunctionSubNode()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc383 StateMap* cond_state_map, const ShapeRefiner& refiner) in Conditional()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc2246 auto refiner = absl::make_unique<SymbolicShapeRefiner>( in InferStatically() local