Searched refs:NodesAcrossFeatureMaps (Results 1 – 4 of 4) sorted by relevance
371 int64 BatchDescriptor::NodesAcrossFeatureMaps() const { in NodesAcrossFeatureMaps() function in stream_executor::dnn::BatchDescriptor381 return input.NodesAcrossFeatureMaps() * output.NodesAcrossFeatureMaps(); in FullyConnectedWeightCount()385 return output.NodesAcrossFeatureMaps(); in FullyConnectedBiasCount()
314 int64 NodesAcrossFeatureMaps() const;
4102 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()4104 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()4149 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()4152 const int ldc = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()4159 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()
3855 const int64 m = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3857 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3902 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3905 const int ldc = output_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()3912 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()