Searched refs:ConvParameters (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv_ops_gpu.h | 29 class FusedConvParameters : public ConvParameters {
37 : ConvParameters(batch, in_depths, in, data_format, out_depths, filter, in FusedConvParameters()
54 return strings::StrCat(ConvParameters::ToString(), ", ", has_side_input_, in ToString()
60 std::tuple<ConvParameters::ParameterDataType, bool, ActivationMode>;
63 return std::make_tuple(ConvParameters::get_data_as_tuple(), has_side_input_, in get_data_as_tuple()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 84 class ConvParameters {
87 ConvParameters(int64 batch, int64 in_depths, const SpatialArray& in, in ConvParameters() function
115 bool operator==(const ConvParameters& other) const {
119 bool operator!=(const ConvParameters& other) const {
|
| D | conv_ops_fused_impl.h | 446 class FusedConvParameters : public ConvParameters {
448 FusedConvParameters(const ConvParameters& base,
450 : ConvParameters(base), activation_mode_(activation_mode) {}
453 return absl::StrCat(ConvParameters::ToString(), ", ", activation_mode_);
461 std::tuple<ConvParameters::ParameterDataType, se::dnn::ActivationMode>;
464 return std::make_tuple(ConvParameters::get_data_as_tuple(),
|
| D | conv_ops_3d.cc | 196 typedef AutoTuneSingleton<Conv3dAutoTuneGroup, ConvParameters,
417 ConvParameters conv_parameters = { in launch()
|
| D | conv_grad_ops_3d.cc | 1085 typedef AutoTuneSingleton<Conv3dBackwardDataAutoTuneGroup, ConvParameters,
1340 const ConvParameters conv_parameters = { in Compute()
1468 typedef AutoTuneSingleton<Conv3dBackwardFilterAutoTuneGroup, ConvParameters,
1746 const ConvParameters conv_parameters = { in Compute()
|
| D | conv_ops.cc | 573 typedef AutoTuneSingleton<ConvAutoTuneGroup, ConvParameters,
828 ConvParameters conv_parameters = { in operator ()()
|
| D | conv_grad_filter_ops.cc | 438 typedef AutoTuneSingleton<ConvBackwardFilterAutoTuneGroup, ConvParameters,
820 ConvParameters conv_parameters = { in operator ()()
|
| D | conv_grad_input_ops.cc | 572 typedef AutoTuneSingleton<ConvBackwardDataAutoTuneGroup, ConvParameters,
932 ConvParameters conv_parameters = { in operator ()()
|
| D | conv_ops_test.cc | 47 ConvParameters params;
50 TEST(ConvParameters, WinogradNonfusedAlgoSize) { in TEST() argument
|