Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc390 .set_input_feature_map_count( in GetGpuConvConfig()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc264 .set_input_feature_map_count(filter_shape.dim_size(2)) in operator ()()
Dconv_ops_3d.cc415 .set_input_feature_map_count(filter_depth) in launch()
Dconv_ops_fused_impl.h581 .set_input_feature_map_count(patch_depths)
Dconv_grad_ops_3d.cc1316 .set_input_feature_map_count(filter_shape.dim_size(3)) in Compute()
1813 .set_input_feature_map_count(filter_shape.dim_size(3)) in Compute()
Dconv_grad_filter_ops.cc869 .set_input_feature_map_count(filter_shape.dim_size(2)) in operator ()()
Dconv_ops.cc894 .set_input_feature_map_count(patch_depths) in operator ()()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h395 FilterDescriptor& set_input_feature_map_count(int64 value) { in set_input_feature_map_count() function