Searched refs:depthwise_multiplier (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_backprop_filter_test.cc | 75 const int64 depthwise_multiplier = use_depth_multiplier ? counter++ : 1; in GetConv2DTestCases() local
77 feature * depthwise_multiplier}; in GetConv2DTestCases()
84 feature * depthwise_multiplier, 1}; in GetConv2DTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 491 const int64 depthwise_multiplier = in HandleConvolution() local
498 new_filter_dimension.push_back(depthwise_multiplier); in HandleConvolution()
532 new_dim->set_size(depthwise_multiplier); in HandleConvolution()
534 new_dim->set_padding_low(depthwise_multiplier - 1); in HandleConvolution()
535 new_dim->set_padding_high(depthwise_multiplier - 1); in HandleConvolution()
547 new_output_dimension.push_back(depthwise_multiplier); in HandleConvolution()
|
| D | hlo_evaluator_typed_visitor.h | 1162 const int64 depthwise_multiplier =
1165 out_index[output_z_dim] / depthwise_multiplier;
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DepthwiseConv2dNativeBackpropInput.pbtxt | 15 `[filter_height, filter_width, in_channels, depthwise_multiplier]`.
|
| D | api_def_DepthwiseConv2dNativeBackpropFilter.pbtxt | 16 `[filter_height, filter_width, in_channels, depthwise_multiplier]` tensor.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 52 int64 depthwise_multiplier = filter_shape.dimensions(output_feature_dim); in GroupedFilterShapeForDepthwiseConvolution() local
59 depthwise_multiplier * input_feature); in GroupedFilterShapeForDepthwiseConvolution()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3091 `[filter_height, filter_width, in_channels, depthwise_multiplier]` tensor.}]>:$filter_sizes,
3123 `[filter_height, filter_width, in_channels, depthwise_multiplier]`.}]>:$filter,
|