Searched refs:kernel_input_feature_dim (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 163 const Shape& filter_shape, int64 kernel_input_feature_dim, in GetExpandedFilterMask() argument
168 ExpandedFilterShape(filter_shape, group_count, kernel_input_feature_dim); in GetExpandedFilterMask()
172 int64 group_size = filter_shape.dimensions(kernel_input_feature_dim); in GetExpandedFilterMask()
183 mask_shape, mask1, {kernel_input_feature_dim})); in GetExpandedFilterMask()
434 int64 kernel_input_feature_dim = dim_numbers.kernel_input_feature_dimension(); in HandleConvolution() local
435 int64 group_size = filter->shape().dimensions(kernel_input_feature_dim); in HandleConvolution()
439 kernel_input_feature_dim); in HandleConvolution()
441 GetExpandedFilterMask(filter->shape(), kernel_input_feature_dim, in HandleConvolution()
459 ShapeUtil::DeleteDimension(kernel_input_feature_dim, filter->shape()); in HandleConvolution()
464 if (i == kernel_input_feature_dim) { in HandleConvolution()
[all …]
|
| D | space_to_batch_converter.cc | 2312 const int64 kernel_input_feature_dim = DimLookUp( in PropagateOnBackpropFilterConv() local
2320 kernel_input_feature_dim); in PropagateOnBackpropFilterConv()
2389 /*new_batch_dim=*/kernel_input_feature_dim, in PropagateOnBackpropFilterConv()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization.cc | 43 auto kernel_input_feature_dim = dnums.kernel_input_feature_dimension(); in Run() local
88 new_kernel_dim_order[num_dims - 2] = kernel_input_feature_dim; in Run()
90 kernel->shape().dimensions(kernel_input_feature_dim); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 184 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension(); in MatchBackwardFilter() local
313 backward_conv_dnums.set_output_batch_dimension(kernel_input_feature_dim); in MatchBackwardFilter()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1105 IntegerAttr kernel_input_feature_dim = in GetConvDimensionNumbersAttr() local
1115 batch_dim, feature_dim, spatial_dims, kernel_input_feature_dim, in GetConvDimensionNumbersAttr()
|