HomeSort by relevance Sort by last modified time
    Searched refs:input_node_info (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transfer_utils.cc 115 for (const std::pair<string, Tensor>& input_node_info : inputs) {
116 const Scope& scope = root.WithOpName(input_node_info.first);
120 .Attr("dtype", input_node_info.second.dtype())
121 .Attr("shape", input_node_info.second.shape());
126 input_types.push_back(input_node_info.second.dtype());
graph_transferer.cc 143 for (const std::pair<string, Tensor>& input_node_info :
147 graph_input_node_info.set_name(input_node_info.first);
148 graph_input_node_info.set_dtype(input_node_info.second.dtype());
149 for (const int64 dim : ToTensorShapeArray(input_node_info.second.shape())) {
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
remote_fused_graph_execute_utils_test.cc 228 std::pair<string, Tensor> input_node_info; local
229 input_node_info.first = NAME_A;
230 input_node_info.second = Tensor(DT_FLOAT, {});
231 input_node_info.second.scalar<float>()() = 1.0f;
232 const std::vector<std::pair<string, Tensor>> inputs{input_node_info};
249 std::pair<string, Tensor> input_node_info; local
250 input_node_info.first = NAME_A;
251 input_node_info.second = Tensor(DT_FLOAT, {});
252 const std::vector<std::pair<string, Tensor>> inputs{input_node_info};
    [all...]
remote_fused_graph_execute_utils.cc 302 for (const std::pair<string, Tensor>& input_node_info :
304 output_tensors.push_back(input_node_info.second);
451 for (const std::pair<string, Tensor>& input_node_info :
453 if (node->name() == input_node_info.first) {
458 input_node_info.second.shape(), &handle);
    [all...]

Completed in 552 milliseconds