Searched defs:BatchDescriptor (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
H A Ddnn.cc212 // -- BatchDescriptor
214 BatchDescriptor::BatchDescriptor(int ndims) function in class:perftools::gputools::dnn::BatchDescriptor
224 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {} function in class:perftools::gputools::dnn::BatchDescriptor
226 std::vector<int64> BatchDescriptor::full_dims(const DataLayout& layout) const {
234 std::vector<int64> BatchDescriptor::full_strides(
253 void BatchDescriptor::CloneFrom(const BatchDescriptor
[all...]
H A Ddnn.h50 // Specify int64 so there's no padding in BatchDescriptor.
81 // Specifies a quantization for activations in a given BatchDescriptor.
208 // BatchDescriptor().set_count(42).set_feature_map_count(7)...
232 class BatchDescriptor { class in namespace:perftools::gputools::dnn
236 BatchDescriptor();
237 explicit BatchDescriptor(int ndims);
240 void CloneFrom(const BatchDescriptor& other);
267 BatchDescriptor& set_count(int64 value) {
271 BatchDescriptor& set_feature_map_count(int64 value) {
275 BatchDescriptor
[all...]

Completed in 132 milliseconds