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 | 167 const Shape& filter_shape, int64_t kernel_input_feature_dim, in GetExpandedFilterMask() argument
172 ExpandedFilterShape(filter_shape, group_count, kernel_input_feature_dim); in GetExpandedFilterMask()
176 int64_t group_size = filter_shape.dimensions(kernel_input_feature_dim); in GetExpandedFilterMask()
187 mask_shape, mask1, {kernel_input_feature_dim})); in GetExpandedFilterMask()
439 int64_t kernel_input_feature_dim = in HandleConvolution() local
441 int64_t group_size = filter->shape().dimensions(kernel_input_feature_dim); in HandleConvolution()
445 kernel_input_feature_dim); in HandleConvolution()
447 GetExpandedFilterMask(filter->shape(), kernel_input_feature_dim, in HandleConvolution()
465 ShapeUtil::DeleteDimension(kernel_input_feature_dim, filter->shape()); in HandleConvolution()
470 if (i == kernel_input_feature_dim) { in HandleConvolution()
[all …]
|
| D | space_to_batch_converter.cc | 3168 const int64_t kernel_input_feature_dim = DimLookUp( in PropagateOnBackpropFilterConv() local
3176 kernel_input_feature_dim); in PropagateOnBackpropFilterConv()
3262 /*new_batch_dim=*/kernel_input_feature_dim, in PropagateOnBackpropFilterConv()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization.cc | 45 auto kernel_input_feature_dim = dnums.kernel_input_feature_dimension(); in Run() local
90 new_kernel_dim_order[num_dims - 2] = kernel_input_feature_dim; in Run()
92 kernel->shape().dimensions(kernel_input_feature_dim); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 105 auto kernel_input_feature_dim = conv_dnums.kernel_input_feature_dimension(); in MatchBackwardFilter() local
234 backward_conv_dnums.set_output_batch_dimension(kernel_input_feature_dim); in MatchBackwardFilter()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 995 int64_t kernel_input_feature_dim = num_spatial_dims; in GetConvDimensionNumbersAttr() local
1006 kernel_input_feature_dim, kernel_output_feature_dim, in GetConvDimensionNumbersAttr()
|