Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor_test.cc82 ShapeRefiner* refiner = nullptr, in ExpectError()
106 ShapeRefiner* refiner = nullptr, in ExpectOK()
1176 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1216 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1279 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1344 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1376 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1397 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1419 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
1444 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, graph_.op_registry()); in TEST_F() local
[all …]
Deval_const_tensor.cc130 const ShapeRefiner& refiner, in TryToInferTensorOutputFromStridedSliceNode()
215 const ShapeRefiner& refiner, in TryToInferTensorOutputFromInputShapes()
295 const Node& target_node, const ShapeRefiner& refiner, in ExtractConstantSubgraph()
499 Status EvaluateConstantTensor(OutputTensor tensor, const ShapeRefiner& refiner, in EvaluateConstantTensor()
Dgraph_constructor.cc173 GraphConstructor(const Options& opts, Graph* g, ShapeRefiner* refiner, in GraphConstructor()
402 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in NodeDefCopyingGraphConstructor()
430 ShapeRefiner* refiner, std::vector<std::pair<Node*, int>>* return_tensors, in NodeDefMovingGraphConstructor()
488 const FunctionDefLibrary* library, Graph* g, ShapeRefiner* refiner, in Construct()
509 const Options& opts, GraphDef&& graph_def, Graph* g, ShapeRefiner* refiner, in Construct()
1453 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph() local
1462 ShapeRefiner refiner(gdef.versions().producer(), g->op_registry()); in ConvertGraphDefToGraph() local
1471 ShapeRefiner refiner(TF_GRAPH_DEF_VERSION, g->op_registry()); in ConvertNodeDefsToGraph() local
1485 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/grappler/costs/
Dgraph_properties.cc2139 SymbolicShapeRefiner* refiner, in ValidateSymbolicShapeManager()
2188 SymbolicShapeRefiner* refiner, in VerboseShapeInferenceLogging()
2605 auto refiner = std::make_unique<SymbolicShapeRefiner>( in InferStatically() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc389 StateMap* cond_state_map, const ShapeRefiner& refiner) in Conditional()