Searched refs:in_depths (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv_ops_gpu.h | 31 FusedConvParameters(int64 batch, int64 in_depths, const SpatialArray& in, in FusedConvParameters() argument
37 : ConvParameters(batch, in_depths, in, data_format, out_depths, filter, in FusedConvParameters()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 87 ConvParameters(int64 batch, int64 in_depths, const SpatialArray& in, in ConvParameters() argument
93 in_depths_(in_depths), in ConvParameters()
104 hash_code_ = Hash64Combine(hash_code_, in_depths); in ConvParameters()
|
| D | conv_ops.cc | 601 const int64 in_depths = GetTensorDim(input, data_format, 'C'); in operator ()() local
609 bool is_grouped_convolution = patch_depths != in_depths; in operator ()()
713 new_in_cols, in_depths), in operator ()()
744 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
745 if (in_depths > 1) { in operator ()()
765 .set_feature_map_count(in_depths) in operator ()()
787 .set_group_count(in_depths / patch_depths); in operator ()()
830 in_depths, // in_depths in operator ()()
|
| D | conv_ops_fused_impl.h | 584 const int64 in_depths = GetTensorDim(input, params.data_format, 'C');
631 new_in_rows, new_in_cols, in_depths),
649 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
650 if (in_depths > 1) {
680 .set_feature_map_count(in_depths)
702 .set_group_count(in_depths / patch_depths);
754 in_depths, // in_depths
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2609 int64 in_depths = input_desc.feature_map_count(); in ShouldIncludeWinogradNonfusedAlgo() local
2615 std::max(in_depths, out_depths) * in_cols * in_rows * in ShouldIncludeWinogradNonfusedAlgo()
|