Lines Matching defs:BatchDescriptor
212 // -- BatchDescriptor
214 BatchDescriptor::BatchDescriptor(int ndims)
224 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {}
226 std::vector<int64> BatchDescriptor::full_dims(const DataLayout& layout) const {
234 std::vector<int64> BatchDescriptor::full_strides(
253 void BatchDescriptor::CloneFrom(const BatchDescriptor& other) {
264 string BatchDescriptor::ToString() const {
276 string BatchDescriptor::ToShortString() const {
313 int64 BatchDescriptor::NodesPerFeatureMap() const {
321 int64 BatchDescriptor::NodesAcrossFeatureMaps() const {
325 int64 BatchDescriptor::ElementCount() const {
329 int64 BatchDescriptor::FullyConnectedWeightCount(
330 const BatchDescriptor& input, const BatchDescriptor& output) {
334 int64 BatchDescriptor::FullyConnectedBiasCount(const BatchDescriptor& output) {
338 BatchDescriptor BatchDescriptor::DepthConcatenateOutputDescriptor(
339 port::ArraySlice<dnn::BatchDescriptor> inputs) {
341 return BatchDescriptor();
347 BatchDescriptor output = inputs[0];