Home
last modified time | relevance | path

Searched refs:NameRangesForNode (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util_test.cc424 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()
Dnode_def_util.h297 Status NameRangesForNode(const NodeDef& node_def, const OpDef& op_def,
299 Status NameRangesForNode(const Node& node, const OpDef& op_def,
Dmemory_types.cc111 TF_RETURN_IF_ERROR(NameRangesForNode(ndef, *op_def, &inp_names, &out_names)); in MemoryTypesForNode()
Dnode_def_util.cc593 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()
Dgraph_to_functiondef.cc210 NameRangesForNode(*node, node->op_def(), nullptr, &output_ranges)); in GraphToFunctionDef()
Dshape_inference.cc230 construction_status_ = NameRangesForNode(*node_def_, op_def, &input_name_map_, in PreInputInit()
Dop_kernel.cc111 NameRangesForNode(*def_, *context->op_def_, &input_name_map_, in OpKernel()
Dfunction.cc1643 TF_CHECK_OK(NameRangesForNode(*n, *op_def, nullptr, &output_names)); in Define()
/external/tensorflow/tensorflow/python/grappler/
Ditem.i271 s = tensorflow::NameRangesForNode(node, *op_def, &inputs, &outputs); in TF_GetColocationGroups()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc402 NameRangesForNode(node_def, *op_def, &input_name_ranges, nullptr)); in CompileTimeConstantInputs()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc157 return NameRangesForNode(node, op_reg_data->op_def, result, in GetInputMap()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc374 NameRangesForNode(*node, node->op_def(), nullptr, &output_ranges)); in GraphToFunctionDef()
Dc_api.cc80 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/
Dfunctions.cc43 TF_RETURN_IF_ERROR(tensorflow::NameRangesForNode( in RegisterFunctionBodyOutputs()