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

  /external/tensorflow/tensorflow/stream_executor/
dnn.cc 96 case DataLayout::kBatchYXDepth:
164 case DataLayout::kBatchYXDepth:
301 case DataLayout::kBatchYXDepth:
dnn.h 54 kBatchYXDepth, // Same as run_brain output, and tensorflow's layout.
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
lrn_op.cc 195 .set_layout(perftools::gputools::dnn::DataLayout::kBatchYXDepth);
412 .set_layout(perftools::gputools::dnn::DataLayout::kBatchYXDepth);
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc 442 case dnn::DataLayout::kBatchYXDepth:
    [all...]

Completed in 190 milliseconds