Home
last modified time | relevance | path

Searched refs:set_output_feature_map_count (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc392 .set_output_feature_map_count( in GetGpuConvConfig()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc265 .set_output_feature_map_count(filter_shape.dim_size(3)) in operator ()()
Dconv_ops_3d.cc416 .set_output_feature_map_count(out_depth) in launch()
Dconv_ops_fused_impl.h582 .set_output_feature_map_count(filter.dim_size(3));
Dconv_grad_ops_3d.cc1317 .set_output_feature_map_count(filter_shape.dim_size(4)) in Compute()
1814 .set_output_feature_map_count(filter_shape.dim_size(4)) in Compute()
Dconv_grad_filter_ops.cc870 .set_output_feature_map_count(filter_shape.dim_size(3)) in operator ()()
Dconv_ops.cc895 .set_output_feature_map_count(filter.dim_size(3)) in operator ()()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h391 FilterDescriptor& set_output_feature_map_count(int64 value) { in set_output_feature_map_count() function