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

  /external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op.cc 77 OpInputList* data_inputs, int* first_dim_size,
101 OP_REQUIRES_OK(c, c->input_list("data", data_inputs));
102 const Tensor& data0 = (*data_inputs)[0];
106 const Tensor& data = (*data_inputs)[input_num];
153 OpInputList data_inputs; variable
157 this->CheckArgsAndAllocateResult(c, &indices_inputs, &data_inputs,
191 auto data_ptr_base = data_inputs[i].template flat<T>().data();
225 OpInputList data_inputs; variable
228 this->CheckArgsAndAllocateResult(c, &indices_inputs, &data_inputs,
245 const Tensor& data = data_inputs[input_num]
    [all...]
  /external/tensorflow/tensorflow/core/framework/
function.cc 345 for (const auto& p : node_info.data_inputs) {
418 nodes_[node_index].data_inputs.push_back(
436 std::vector<std::pair<int, int>> data_inputs; member in struct:tensorflow::__anon39561::FunctionInstantiationHelper::NodeInfo
    [all...]

Completed in 72 milliseconds