Searched refs:NodesAcrossFeatureMaps (Results 1 – 4 of 4) sorted by relevance
340 int64 BatchDescriptor::NodesAcrossFeatureMaps() const { in NodesAcrossFeatureMaps() function in stream_executor::dnn::BatchDescriptor350 return input.NodesAcrossFeatureMaps() * output.NodesAcrossFeatureMaps(); in FullyConnectedWeightCount()354 return output.NodesAcrossFeatureMaps(); in FullyConnectedBiasCount()
322 int64 NodesAcrossFeatureMaps() const;
3569 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3571 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3616 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3619 const int ldc = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3626 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()
3329 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3331 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3376 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3379 const int ldc = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3386 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()