Searched refs:data_inputs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op.cc | 77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult() argument
101 OP_REQUIRES_OK(c, c->input_list("data", data_inputs)); in CheckArgsAndAllocateResult()
102 const Tensor& data0 = (*data_inputs)[0]; in CheckArgsAndAllocateResult()
106 const Tensor& data = (*data_inputs)[input_num]; in CheckArgsAndAllocateResult()
153 OpInputList data_inputs; in Compute() local
157 this->CheckArgsAndAllocateResult(c, &indices_inputs, &data_inputs, in Compute()
191 auto data_ptr_base = data_inputs[i].template flat<T>().data(); in Compute()
225 OpInputList data_inputs; in Compute() local
228 this->CheckArgsAndAllocateResult(c, &indices_inputs, &data_inputs, in Compute()
245 const Tensor& data = data_inputs[input_num]; in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.cc | 345 for (const auto& p : node_info.data_inputs) { in AddNodeInputs()
418 nodes_[node_index].data_inputs.push_back( in AddInput()
436 std::vector<std::pair<int, int>> data_inputs; member
|