Searched refs:feature_map_count (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 282 bdyx_dims[1] = feature_map_count(); in full_dims()
322 count(), feature_map_count(), spatial, value_min_, value_max_, in ToString()
330 std::string depth = absl::StrCat("d", feature_map_count()); in ToShortString()
372 return NodesPerFeatureMap() * feature_map_count(); in NodesAcrossFeatureMaps()
376 return count() * feature_map_count() * NodesPerFeatureMap(); in ElementCount()
393 int feature_map_count = 0; in DepthConcatenateOutputDescriptor() local
395 feature_map_count += dimensions.feature_map_count(); in DepthConcatenateOutputDescriptor()
398 output.set_feature_map_count(feature_map_count); in DepthConcatenateOutputDescriptor()
|
| D | dnn.h | 251 int64 feature_map_count() const { return tensor_.dimensions(1); } in feature_map_count() function
|
| D | stream.cc | 1083 input_dimensions[i].feature_map_count() != in ThenSpaceConcatenate()
1084 input_dimensions[0].feature_map_count())) { in ThenSpaceConcatenate()
1096 input_dimensions[i].feature_map_count() != in ThenSpaceConcatenate()
1097 input_dimensions[0].feature_map_count())) { in ThenSpaceConcatenate()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 544 batch_descriptor.count(), batch_descriptor.feature_map_count(), in CudnnTensorDescriptor()
4140 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
4147 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
4160 output_dimensions.feature_map_count(); in DoMatMul()
4168 const int output_offset = i * output_dimensions.feature_map_count(); in DoMatMul()
4202 .set_feature_map_count(dimensions.feature_map_count()) in DoBiasAdd()
4563 for (int64 depth = 0; depth < dimensions.feature_map_count(); ++depth) { in DoDepthConcatenate()
4567 output_dimensions.feature_map_count())] = in DoDepthConcatenate()
4568 tmp[index(batch, depth, yx, dimensions.feature_map_count())]; in DoDepthConcatenate()
4572 depth_sum += dimensions.feature_map_count(); in DoDepthConcatenate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.cc | 105 .set_feature_map_count(batch_descs.input_desc.feature_map_count()) in MakeBatchNormDescriptors()
|
| D | gpu_conv_runner.cc | 108 params.config.output_descriptor.feature_map_count()) in RunGpuConvForwardActivation()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3893 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
3900 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
3913 output_dimensions.feature_map_count(); in DoMatMul()
3921 const int output_offset = i * output_dimensions.feature_map_count(); in DoMatMul()
3955 .set_feature_map_count(dimensions.feature_map_count()) in DoBiasAdd()
4523 for (int64 depth = 0; depth < dimensions.feature_map_count(); ++depth) { in DoDepthConcatenate()
4527 output_dimensions.feature_map_count())] = in DoDepthConcatenate()
4528 tmp[index(batch, depth, yx, dimensions.feature_map_count())]; in DoDepthConcatenate()
4532 depth_sum += dimensions.feature_map_count(); in DoDepthConcatenate()
|