Home
last modified time | relevance | path

Searched defs:out_depths (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv_ops_gpu.h32 TensorFormat data_format, int64 out_depths, in FusedConvParameters()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h88 TensorFormat data_format, int64 out_depths, in ConvParameters()
Dconv_ops.cc670 const int64 out_depths = GetTensorDim(*output, data_format, 'C'); in operator ()() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2612 int64 out_depths = output_desc.feature_map_count(); in ShouldIncludeWinogradNonfusedAlgo() local