Searched refs:node_output (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 30 : node_name(node_name), node_output(output), position(position) { in FunctionDefTensorDesc()
31 full_str = strings::StrCat(node_name, ":", node_output, ":", position); in FunctionDefTensorDesc()
55 node_output = string(capture.data(), capture.size()); in FunctionDefTensorDesc()
|
| D | function_utils.h | 52 string node_output; member
|
| D | function_utils_test.cc | 34 EXPECT_EQ(f.node_output, "y"); in TEST()
40 EXPECT_EQ(f2.node_output, ""); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 109 string node_output; in ExpandFunctionDefInput() local
130 node_output = string(capture.data(), capture.size()); in ExpandFunctionDefInput()
143 bool is_function_body_output = !node_output.empty(); in ExpandFunctionDefInput()
176 auto output = outputs.find(node_output); in ExpandFunctionDefInput()
191 "Invalid node ", node_name, " output ", node_output, in ExpandFunctionDefInput()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.cc | 155 NodeOutput* node_output = stats_->add_output(); in SetOutput() local
156 node_output->set_slot(slot); in SetOutput()
157 tensor->FillDescription(node_output->mutable_tensor_description()); in SetOutput()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 465 for (const auto& node_output : node_stat.output()) { in GetStatsStringFromRunMetadata() local
468 const auto size = node_output.tensor_description() in GetStatsStringFromRunMetadata()
|