Searched refs:NameRangesForNode (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util_test.cc | 424 TF_EXPECT_OK(NameRangesForNode(node_def, op_def, &inputs, &outputs)); in TEST() 432 EXPECT_FALSE(NameRangesForNode(node_def, bad_op_def, &inputs, &outputs).ok()); in TEST() 445 TF_EXPECT_OK(NameRangesForNode(node_def1, op_def, &inputs, &outputs)); in TEST() 454 TF_EXPECT_OK(NameRangesForNode(node_def2, op_def, &inputs, &outputs)); in TEST() 476 TF_EXPECT_OK(NameRangesForNode(node_def1, op_def, &inputs, &outputs)); in TEST() 489 TF_EXPECT_OK(NameRangesForNode(node_def2, op_def, &inputs, &outputs)); in TEST() 498 EXPECT_FALSE(NameRangesForNode(bad_node_def, op_def, &inputs, &outputs).ok()); in TEST() 517 TF_EXPECT_OK(NameRangesForNode(node_def1, op_def, &inputs, &outputs)); in TEST() 531 TF_EXPECT_OK(NameRangesForNode(node_def2, op_def, &inputs, &outputs)); in TEST() 544 EXPECT_FALSE(NameRangesForNode(bad_node_def, op_def, &inputs, &outputs).ok()); in TEST()
|
D | node_def_util.h | 297 Status NameRangesForNode(const NodeDef& node_def, const OpDef& op_def, 299 Status NameRangesForNode(const Node& node, const OpDef& op_def,
|
D | memory_types.cc | 111 TF_RETURN_IF_ERROR(NameRangesForNode(ndef, *op_def, &inp_names, &out_names)); in MemoryTypesForNode()
|
D | node_def_util.cc | 593 Status NameRangesForNode(const NodeDef& node_def, const OpDef& op_def, in NameRangesForNode() function 605 Status NameRangesForNode(const Node& node, const OpDef& op_def, in NameRangesForNode() function 607 return NameRangesForNode(node.def(), op_def, inputs, outputs); in NameRangesForNode()
|
D | graph_to_functiondef.cc | 210 NameRangesForNode(*node, node->op_def(), nullptr, &output_ranges)); in GraphToFunctionDef()
|
D | shape_inference.cc | 230 construction_status_ = NameRangesForNode(*node_def_, op_def, &input_name_map_, in PreInputInit()
|
D | op_kernel.cc | 111 NameRangesForNode(*def_, *context->op_def_, &input_name_map_, in OpKernel()
|
D | function.cc | 1643 TF_CHECK_OK(NameRangesForNode(*n, *op_def, nullptr, &output_names)); in Define()
|
/external/tensorflow/tensorflow/python/grappler/ |
D | item.i | 271 s = tensorflow::NameRangesForNode(node, *op_def, &inputs, &outputs); in TF_GetColocationGroups()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_op_registry.cc | 402 NameRangesForNode(node_def, *op_def, &input_name_ranges, nullptr)); in CompileTimeConstantInputs()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | map_vectorization.cc | 157 return NameRangesForNode(node, op_reg_data->op_def, result, in GetInputMap()
|
/external/tensorflow/tensorflow/c/ |
D | c_api_function.cc | 374 NameRangesForNode(*node, node->op_def(), nullptr, &output_ranges)); in GraphToFunctionDef()
|
D | c_api.cc | 80 using tensorflow::NameRangesForNode; 1526 NameRangesForNode(oper->node, oper->node.op_def(), nullptr, &name_ranges); in TF_OperationOutputListLength() 1548 NameRangesForNode(oper->node, oper->node.op_def(), &name_ranges, nullptr); in TF_OperationInputListLength()
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | functions.cc | 43 TF_RETURN_IF_ERROR(tensorflow::NameRangesForNode( in RegisterFunctionBodyOutputs()
|