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

  /external/tensorflow/tensorflow/contrib/lite/
graph_info.cc 140 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) {
141 if (tensor_epochs_[input_tensor_index] == kEpochNotReady) {
164 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) {
165 int input_epoch = tensor_epochs_[input_tensor_index];
168 current_subgraph.input_tensors.push_back(input_tensor_index);
174 input_subgraph.output_tensors.push_back(input_tensor_index);
  /external/tensorflow/tensorflow/core/kernels/
conv_ops_gpu_3.cu.cc 188 Index<3> input_tensor_index; local
189 input_tensor_index[0] = output_tensor_index[2];
190 input_tensor_index[1] = output_tensor_index[1];
191 input_tensor_index[2] = output_tensor_index[0];
193 int input_index = TensorIndexToFlat(input_tensor_index, input_dims);
214 Index<3> input_tensor_index; local
215 input_tensor_index[0] = output_tensor_index[0];
216 input_tensor_index[1] = output_tensor_index[2];
217 input_tensor_index[2] = output_tensor_index[1];
219 int input_index = TensorIndexToFlat(input_tensor_index, input_dims)
383 Index<NDIMS> input_tensor_index; local
412 Index<NDIMS> input_tensor_index; local
    [all...]

Completed in 131 milliseconds