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

  /external/tensorflow/tensorflow/contrib/lite/
graph_info.cc 77 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) {
78 tensor_epochs_[output_tensor_index] = kEpochNotReady;
159 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) {
160 tensor_epochs_[output_tensor_index] = current_epoch;
  /external/tensorflow/tensorflow/core/kernels/
conv_ops_gpu_3.cu.cc 186 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims); 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];
212 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims); 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];
380 Index<NDIMS> output_tensor_index = local
384 input_tensor_index[0] = output_tensor_index[0]; // batc
409 Index<NDIMS> output_tensor_index = local
    [all...]

Completed in 200 milliseconds