| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| D | call.cc | 60 TfLiteTensor* node_input = context->tensors + node->inputs->data[i]; in ValidateAndResizeInputsIfNeeded() local
63 TF_LITE_ENSURE_TYPES_EQ(context, node_input->type, subgraph_input->type); in ValidateAndResizeInputsIfNeeded()
65 context, node_input->dims->size > 0, in ValidateAndResizeInputsIfNeeded()
68 TF_LITE_ENSURE_EQ(context, node_input->dims->data[0], loop_count); in ValidateAndResizeInputsIfNeeded()
72 new_dims.reserve(node_input->dims->size); in ValidateAndResizeInputsIfNeeded()
74 new_dims.insert(new_dims.end(), node_input->dims->data + 1, in ValidateAndResizeInputsIfNeeded()
75 node_input->dims->data + node_input->dims->size); in ValidateAndResizeInputsIfNeeded()
79 if (!MatchDimensionsExceptBatchSize(node_input, subgraph_input)) { in ValidateAndResizeInputsIfNeeded()
81 for (int i = 0; i < node_input->dims->size; i++) { in ValidateAndResizeInputsIfNeeded()
82 node_input_dims << node_input->dims->data[i] << " "; in ValidateAndResizeInputsIfNeeded()
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_stats.cc | 211 string node_input = it->second->input(i); in AddGraph() local
215 auto prefix_pos = node_input.find(':'); in AddGraph()
216 if (prefix_pos != node_input.npos) { in AddGraph()
217 std::vector<string> input_parts = absl::StrSplit(node_input, ':'); in AddGraph()
219 << "Unknown NodeDef.input format: " << node_input; in AddGraph()
220 node_input = input_parts[0]; in AddGraph()
224 if (node_input.substr(0, 1) == "^") { in AddGraph()
225 node_input = node_input.substr(1); in AddGraph()
230 node->AddInput(node_input, output_idx, i); in AddGraph()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 45 for (const std::string& node_input : node.input()) { in AddNodeToGraph() local
48 if (StrContains(node_input, cluster_name) && in AddNodeToGraph()
56 new_node->add_input(node_input); in AddNodeToGraph()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 196 for (auto& node_input : *function_node.mutable_input()) { in GetUniqueSignature()
197 const auto& input = ParseNodeConnection(node_input); in GetUniqueSignature()
200 node_input = *new_name + ParseOutputNode(node_input); in GetUniqueSignature()
217 for (auto& node_input : *function_node.mutable_input()) { in FuseFunctionNodes()
218 auto parsed_name = ParseNodeConnection(node_input); in FuseFunctionNodes()
225 node_input = in FuseFunctionNodes()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_test.py | 142 for (input_index, node_input) in enumerate(node.input):
143 if node_input == "b":
145 if node_input == "y":
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | graph_util_impl.py | 381 for node_input in node.input:
382 if "^" in node_input:
383 control_input_names.add(node_input.replace("^", ""))
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_transfer_info.proto | 35 repeated GraphTransferNodeInput node_input = 2; field
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.cc | 921 const string& node_input = node_def->input(i); in AddControlDependencies() local
922 if (node_input[0] != '^') { in AddControlDependencies()
927 if (node_input == input) { in AddControlDependencies()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2095 std::vector<NodeInput> node_input{ in TEST_P() local
2139 for (int i = 0; i < node_input.size(); i++) { in TEST_P()
2160 node_input[i].name, in TEST_P()
2169 partial_input_shape[1] = node_input[i].dims[1]; in TEST_P()
2171 AddTestTensor(node_input[i].name, node_input[i].dims, tf_type_, in TEST_P()
2172 node_input[i].val, partial_input_shape, in TEST_P()
2176 AddTestWeights(node_input[i].name, node_input[i].dims, in TEST_P()
2177 node_input[i].val, tf_type_); in TEST_P()
2180 TestOpConverter("my_batchnorm", node_def, node_input[0].dims, in TEST_P()
|
| /external/ComputeLibrary/examples/ |
| D | graph_edsr.h | 685 INode *node_input = _graph.node(id_input); in setup() local
686 node_input->set_common_node_parameters(NodeParams{ "input", target }); in setup()
687 node_input->output(0)->set_accessor(get_input_accessor(common_params)); in setup()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision.cc | 1863 GraphView::InputPort node_input(&node, 0); in NodeImplicitlyReadsNonResourceVariable() local
1865 graph_view_.GetRegularFanin(node_input); in NodeImplicitlyReadsNonResourceVariable()
|