Home
last modified time | relevance | path

Searched refs:set_feature_map_count (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc75 .set_feature_map_count(in_features) in Compute()
79 .set_feature_map_count(in_features) in Compute()
209 .set_feature_map_count(in_features) in Compute()
214 .set_feature_map_count(in_features) in Compute()
Dpooling_ops_common.cc366 .set_feature_map_count(depth) in Compute()
373 .set_feature_map_count(depth) in Compute()
681 .set_feature_map_count(params.depth) in Compute()
688 .set_feature_map_count(params.depth) in Compute()
Dlrn_op.cc207 .set_feature_map_count(depth) in launch()
260 .set_feature_map_count(depth) in launch()
499 .set_feature_map_count(depth) in launch()
574 .set_feature_map_count(depth) in launch()
Dconv_ops_fused_impl.h574 .set_feature_map_count(in_depths)
587 .set_feature_map_count(out_depths)
601 .set_feature_map_count(out_depths)
Dconv_grad_input_ops.cc253 .set_feature_map_count(dims.in_depth) in operator ()()
259 .set_feature_map_count(dims.out_depth) in operator ()()
Dconv_ops_3d.cc399 .set_feature_map_count(in_depth) in launch()
409 .set_feature_map_count(out_depth) in launch()
Dfused_batch_norm_op.cc892 .set_feature_map_count(channels) in operator ()()
899 .set_feature_map_count(channels) in operator ()()
1083 .set_feature_map_count(channels) in operator ()()
1090 .set_feature_map_count(channels) in operator ()()
Dconv_grad_ops_3d.cc1303 .set_feature_map_count(dims.in_depth) in Compute()
1310 .set_feature_map_count(dims.out_depth) in Compute()
1800 .set_feature_map_count(dims.in_depth) in Compute()
1807 .set_feature_map_count(dims.out_depth) in Compute()
Dconv_grad_filter_ops.cc858 .set_feature_map_count(dims.in_depth) in operator ()()
864 .set_feature_map_count(dims.out_depth) in operator ()()
Dconv_ops.cc881 .set_feature_map_count(in_depths) in operator ()()
889 .set_feature_map_count(out_depths) in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc100 .set_feature_map_count(shape.dimensions(feature_index)) in MakeBatchNormDescriptors()
105 .set_feature_map_count(batch_descs.input_desc.feature_map_count()) in MakeBatchNormDescriptors()
Dgpu_conv_runner.cc107 .set_feature_map_count( in RunGpuConvForwardActivation()
377 .set_feature_map_count( in GetGpuConvConfig()
419 .set_feature_map_count( in GetGpuConvConfig()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc398 output.set_feature_map_count(feature_map_count); in DepthConcatenateOutputDescriptor()
Ddnn.h275 BatchDescriptor& set_feature_map_count(int64 value) { in set_feature_map_count() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4202 .set_feature_map_count(dimensions.feature_map_count()) in DoBiasAdd()
4637 .set_feature_map_count(dims[1]) in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3955 .set_feature_map_count(dimensions.feature_map_count()) in DoBiasAdd()
4604 .set_feature_map_count(dims[1]) in DeriveOutputBatchDescriptor()