Searched refs:input_feature_dim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization.cc | 42 auto input_feature_dim = dnums.input_feature_dimension(); in Run() local
69 new_input_dim_order[num_dims - 1] = input_feature_dim; in Run()
71 input->shape().dimensions(input_feature_dim); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 104 int64 input_feature_dim) { in ExpandedFilterShape() argument
109 input_feature_dim, shape.dimensions(input_feature_dim) * group_count); in ExpandedFilterShape()
|
| D | dynamic_padder.cc | 906 const int64 input_feature_dim = dnums.input_feature_dimension(); in RewriteDynamicConvolutionForward() local
909 custom_call_conv->mutable_operand(0), {}, input_feature_dim)) { in RewriteDynamicConvolutionForward()
910 input = PadWithScalar(input, input_feature_dim, input_feature_dynamic_size, in RewriteDynamicConvolutionForward()
989 const int64 input_feature_dim = dnums.input_feature_dimension(); in RewriteDynamicConvolutionKernelGrad() local
992 custom_call_conv->mutable_operand(0), {}, input_feature_dim)) { in RewriteDynamicConvolutionKernelGrad()
993 activations = PadWithScalar(activations, input_feature_dim, in RewriteDynamicConvolutionKernelGrad()
|
| D | hlo_cost_analysis.cc | 580 const int64 input_feature_dim = dnums.input_feature_dimension(); in HandleConvolution() local
583 ShapeUtil::GetDimension(lhs_shape, input_feature_dim); in HandleConvolution()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 50 int64 input_feature_dim = filter_shape.dimensions_size() - 2; in GroupedFilterShapeForDepthwiseConvolution() local
53 int64 input_feature = filter_shape.dimensions(input_feature_dim); in GroupedFilterShapeForDepthwiseConvolution()
57 grouped_filter_shape.set_dimensions(input_feature_dim, 1); in GroupedFilterShapeForDepthwiseConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 182 auto input_feature_dim = conv_dnums.input_feature_dimension(); in MatchBackwardFilter() local
308 backward_conv_dnums.set_input_batch_dimension(input_feature_dim); in MatchBackwardFilter()
|