Home
last modified time | relevance | path

Searched refs:feature_map_count (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc251 bdyx_dims[1] = feature_map_count(); in full_dims()
291 count(), feature_map_count(), spatial.c_str(), value_min_, value_max_, in ToString()
299 string depth = absl::StrCat("d", feature_map_count()); in ToShortString()
341 return NodesPerFeatureMap() * feature_map_count(); in NodesAcrossFeatureMaps()
345 return count() * feature_map_count() * NodesPerFeatureMap(); in ElementCount()
362 int feature_map_count = 0; in DepthConcatenateOutputDescriptor() local
364 feature_map_count += dimensions.feature_map_count(); in DepthConcatenateOutputDescriptor()
367 output.set_feature_map_count(feature_map_count); in DepthConcatenateOutputDescriptor()
Ddnn.h259 int64 feature_map_count() const { return tensor_.dimensions(1); }
Dstream.cc1508 input_dimensions[i].feature_map_count() != in ThenSpaceConcatenate()
1509 input_dimensions[0].feature_map_count())) { in ThenSpaceConcatenate()
1521 input_dimensions[i].feature_map_count() != in ThenSpaceConcatenate()
1522 input_dimensions[0].feature_map_count())) { in ThenSpaceConcatenate()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc522 batch_descriptor.count(), batch_descriptor.feature_map_count(), in CudnnTensorDescriptor()
2609 int64 in_depths = input_desc.feature_map_count(); in ShouldIncludeWinogradNonfusedAlgo()
2612 int64 out_depths = output_desc.feature_map_count(); in ShouldIncludeWinogradNonfusedAlgo()
2744 SE_RETURN_IF_ERROR(check_sizes(input_descriptor.feature_map_count(), in DoConvolve()
2745 output_descriptor.feature_map_count())); in DoConvolve()
2747 input_descriptor.feature_map_count())); in DoConvolve()
2749 output_descriptor.feature_map_count())); in DoConvolve()
3607 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
3614 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
3627 output_dimensions.feature_map_count(); in DoMatMul()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc70 .set_feature_map_count(input_desc.feature_map_count()) in MakeDescriptors()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3367 output_dimensions.feature_map_count() == 1)) { in DoMatMul()
3374 const uint64 m = output_dimensions.feature_map_count(); in DoMatMul()
3387 output_dimensions.feature_map_count(); in DoMatMul()
3395 const int output_offset = i * output_dimensions.feature_map_count(); in DoMatMul()
3429 .set_feature_map_count(dimensions.feature_map_count()) in DoBiasAdd()
3926 for (int64 depth = 0; depth < dimensions.feature_map_count(); ++depth) { in DoDepthConcatenate()
3930 output_dimensions.feature_map_count())] = in DoDepthConcatenate()
3931 tmp[index(batch, depth, yx, dimensions.feature_map_count())]; in DoDepthConcatenate()
3935 depth_sum += dimensions.feature_map_count(); in DoDepthConcatenate()