Searched refs:out_depths (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv_ops_gpu.h | 32 TensorFormat data_format, int64 out_depths, 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 | 88 TensorFormat data_format, int64 out_depths, in ConvParameters() argument
94 out_depths_(out_depths), in ConvParameters()
107 hash_code_ = Hash64Combine(hash_code_, out_depths); in ConvParameters()
|
| D | conv_ops_fused_impl.h | 595 const int64 out_depths = GetTensorDim(*output, params.data_format, 'C');
602 OP_REQUIRES(context, bias.dim_size(0) == out_depths,
693 .set_feature_map_count(out_depths)
707 .set_feature_map_count(out_depths)
729 out_cols, out_depths),
758 out_depths, // out_depths
|
| D | conv_ops.cc | 670 const int64 out_depths = GetTensorDim(*output, data_format, 'C'); in operator ()() local
773 .set_feature_map_count(out_depths) in operator ()()
806 out_rows, out_cols, out_depths), in operator ()()
834 out_depths, // out_depths in operator ()()
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cCopyImageTests.hpp | 130 glw::GLuint* out_widths, glw::GLuint* out_heights, glw::GLuint* out_depths) const;
|
| D | gl4cCopyImageTests.cpp | 2671 GLuint* out_heights, GLuint* out_depths) const in calculateDimmensions()
2699 out_depths[i] = FUNCTIONAL_TEST_N_LAYERS * factors[i] / divide; in calculateDimmensions()
2703 out_depths[i] = n_layers; in calculateDimmensions()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2612 int64 out_depths = output_desc.feature_map_count(); in ShouldIncludeWinogradNonfusedAlgo() local
2615 std::max(in_depths, out_depths) * in_cols * in_rows * in ShouldIncludeWinogradNonfusedAlgo()
|