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

  /external/tensorflow/tensorflow/stream_executor/
dnn.cc 340 int64 BatchDescriptor::NodesAcrossFeatureMaps() const {
350 return input.NodesAcrossFeatureMaps() * output.NodesAcrossFeatureMaps();
354 return output.NodesAcrossFeatureMaps();
dnn.h 322 int64 NodesAcrossFeatureMaps() const;
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.cc     [all...]

Completed in 291 milliseconds