| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util_test.cc | 490 TF_EXPECT_OK(NameRangesForNode(node_def, op_def, &inputs, &outputs)); in TEST()
498 EXPECT_FALSE(NameRangesForNode(node_def, bad_op_def, &inputs, &outputs).ok()); in TEST()
512 TF_EXPECT_OK(NameRangesForNode(node_def1, op_def, &inputs, &outputs)); in TEST()
522 TF_EXPECT_OK(NameRangesForNode(node_def2, op_def, &inputs, &outputs)); in TEST()
545 TF_EXPECT_OK(NameRangesForNode(node_def1, op_def, &inputs, &outputs)); in TEST()
559 TF_EXPECT_OK(NameRangesForNode(node_def2, op_def, &inputs, &outputs)); in TEST()
568 EXPECT_FALSE(NameRangesForNode(bad_node_def, op_def, &inputs, &outputs).ok()); in TEST()
587 TF_EXPECT_OK(NameRangesForNode(node_def1, op_def, &inputs, &outputs)); in TEST()
602 TF_EXPECT_OK(NameRangesForNode(node_def2, op_def, &inputs, &outputs)); in TEST()
615 EXPECT_FALSE(NameRangesForNode(bad_node_def, op_def, &inputs, &outputs).ok()); in TEST()
|
| D | memory_types.cc | 120 NameRangesForNode(ndef, *op_def, &inp_names, &out_names)); in MemoryTypesForNode()
|
| D | node_def_util.h | 365 Status NameRangesForNode(const AttrSlice& attrs, const OpDef& op_def,
|
| D | graph_to_functiondef.cc | 484 NameRangesForNode(*node, node->op_def(), nullptr, &output_ranges)); in GraphToFunctionDef()
|
| D | op_kernel.cc | 112 NameRangesForNode(props_->node_def, *props_->op_def, in OpKernel()
138 NameRangesForNode(props_->node_def, *props_->op_def, in OpKernel()
|
| D | node_def_util.cc | 711 Status NameRangesForNode(const AttrSlice& attrs, const OpDef& op_def, in NameRangesForNode() function
|
| D | shape_inference.cc | 173 NameRangesForNode(attrs_, op_def, &input_name_map_, &output_name_map_); in PreInputInit()
|
| D | function.cc | 1918 TF_CHECK_OK(NameRangesForNode(*n, *op_def, nullptr, &output_names)); in Define()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_node_util.cc | 37 Status NameRangesForNode(const Node& node, const OpDef& op_def, in NameRangesForNode() function
39 return NameRangesForNode(node.def(), op_def, inputs, outputs); in NameRangesForNode()
|
| D | graph_node_util.h | 53 Status NameRangesForNode(const Node& node, const OpDef& op_def,
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation.cc | 231 NameRangesForNode(AttrSlice(&attrs), *op_def, &name_ranges, nullptr)); in InputLength()
258 NameRangesForNode(AttrSlice(&attrs), *op_def, nullptr, &name_ranges)); in OutputLength()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | item_wrapper.cc | 225 if (!tensorflow::NameRangesForNode(node, *op_def, &inputs, &outputs) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 161 return NameRangesForNode(node, op_reg_data->op_def, result, in GetInputMap()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.cc | 423 NameRangesForNode(node_def, *op_def, &input_name_ranges, nullptr)); in CompileTimeConstantInputs()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 457 TF_RETURN_IF_ERROR(tensorflow::NameRangesForNode( in Initialize()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 84 using tensorflow::NameRangesForNode;
1105 NameRangesForNode(oper->node, oper->node.op_def(), nullptr, &name_ranges); in TF_OperationOutputListLength()
1127 NameRangesForNode(oper->node, oper->node.op_def(), &name_ranges, nullptr); in TF_OperationInputListLength()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1817 NameRangesForNode(node, node.op_def(), &input_ranges, &output_ranges)); in CreateOperation()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 4001 TF_RETURN_IF_ERROR(NameRangesForNode(replicate_node, replicate_node.op_def(), in GetIOTypes()
|