Home
last modified time | relevance | path

Searched refs:feature_group_count (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter.cc43 int64 feature_group_count, in CreateCudnnConv() argument
62 custom_call->set_feature_group_count(feature_group_count); in CreateCudnnConv()
96 if (conv->feature_group_count() > 1) { in MatchBackwardFilter()
266 if (conv->feature_group_count() > 1) { in MatchBackwardInput()
460 if (conv->feature_group_count() == 1) { in MatchBackwardInput()
483 input_features / conv->feature_group_count()); in MatchBackwardInput()
485 output_features * conv->feature_group_count()); in MatchBackwardInput()
511 window, dnums, conv->feature_group_count(), in RunOnInstruction()
519 conv->feature_group_count(), conv->metadata()); in RunOnInstruction()
528 conv->feature_group_count(), conv->metadata()); in RunOnInstruction()
Dcudnn_conv_runner.cc71 int64 feature_group_count; member
126 int64 feature_group_count = params.feature_group_count; in RunCudnnConvImpl() local
206 convolution_descriptor.set_group_count(feature_group_count); in RunCudnnConvImpl()
336 params.feature_group_count = conv->feature_group_count(); in GetCudnnConvParams()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_conv_op.cc50 int64 feature_group_count; in Compile() local
58 "feature_group_count", &feature_group_count)); in Compile()
79 lhs_dilation, rhs_dilation, dnums_, feature_group_count, in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter_test.cc58 EXPECT_EQ(root->feature_group_count(), 1); in TEST_F()
93 EXPECT_EQ(root->operand(0)->feature_group_count(), 1); in TEST_F()
94 EXPECT_EQ(root->operand(1)->feature_group_count(), 1); in TEST_F()
Dhlo_creation_utils.cc83 HloInstruction* lhs, HloInstruction* rhs, int64 feature_group_count, in MakeConvolveHlo() argument
90 lhs->shape(), rhs->shape(), feature_group_count, 1, in MakeConvolveHlo()
93 convolve_shape, lhs, rhs, feature_group_count, 1, window, in MakeConvolveHlo()
Dshape_inference.cc1572 const Shape& lhs, const Shape& rhs, int64 feature_group_count, in InferConvolveShape() argument
1578 if (feature_group_count <= 0) { in InferConvolveShape()
1581 feature_group_count); in InferConvolveShape()
1590 if (batch_group_count > 1 && feature_group_count > 1) { in InferConvolveShape()
1594 batch_group_count, feature_group_count); in InferConvolveShape()
1721 if (input_features % feature_group_count != 0 || in InferConvolveShape()
1722 input_features / feature_group_count != kernel_input_features) { in InferConvolveShape()
1729 input_features, feature_group_count, kernel_input_features, in InferConvolveShape()
1734 if (kernel_output_features % feature_group_count > 0) { in InferConvolveShape()
1746 kernel_output_features, feature_group_count, in InferConvolveShape()
Dhlo_creation_utils.h58 HloInstruction* lhs, HloInstruction* rhs, int64 feature_group_count,
Dhlo_instructions.h1033 int64 feature_group_count, int64 batch_group_count, const Window& window,
1047 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count() function
1203 void set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() argument
1204 feature_group_count_ = feature_group_count; in set_feature_group_count()
1209 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count() function
Dtranspose_folding.cc178 convolution.shape(), new_lhs, new_rhs, convolution.feature_group_count(), in FoldTransposeIntoConvolution()
Dhlo_instruction.cc423 std::max<int64>(proto.feature_group_count(), 1), in CreateFromProto()
472 std::max(static_cast<int64>(proto.feature_group_count()), 1LL)); in CreateFromProto()
782 int64 feature_group_count, int64 batch_group_count, const Window& window, in CreateConvolve() argument
786 shape, lhs, rhs, feature_group_count, batch_group_count, window, in CreateConvolve()
3460 int64 HloInstruction::feature_group_count() const { in feature_group_count() function in xla::HloInstruction
3462 return convolution->feature_group_count(); in feature_group_count()
3464 return Cast<HloCustomCallInstruction>(this)->feature_group_count(); in feature_group_count()
3467 void HloInstruction::set_feature_group_count(int64 feature_group_count) { in set_feature_group_count() argument
3469 feature_group_count); in set_feature_group_count()
Dhlo_instruction.h439 int64 feature_group_count, int64 batch_group_count, const Window& window,
1559 int64 feature_group_count() const;
1561 void set_feature_group_count(int64 feature_group_count);
Dshape_inference.h111 const Shape& lhs, const Shape& rhs, int64 feature_group_count,
Dhlo_parser.cc1067 optional<int64> feature_group_count; in ParseInstructionRhs() local
1073 &feature_group_count}; in ParseInstructionRhs()
1086 if (!feature_group_count) { in ParseInstructionRhs()
1087 feature_group_count = 1; in ParseInstructionRhs()
1102 feature_group_count.value(), batch_group_count.value(), *window, in ParseInstructionRhs()
1488 optional<int64> feature_group_count; in ParseInstructionRhs() local
1498 &feature_group_count}; in ParseInstructionRhs()
1553 if (feature_group_count.has_value()) { in ParseInstructionRhs()
1554 instruction->set_feature_group_count(*feature_group_count); in ParseInstructionRhs()
Dhlo_evaluator_typed_visitor.h1039 lhs_shape, rhs_shape, conv->feature_group_count(), in HandleConvolution()
1063 const int64 feature_group_count = conv->feature_group_count(); in HandleConvolution() local
1068 rhs_literal_data, feature_group_count, in HandleConvolution()
1089 const int64 input_feature_group_size = input_z_size / feature_group_count; in HandleConvolution()
1096 output_z_size / feature_group_count; in HandleConvolution()
Dconvolution_group_converter.cc328 int64 group_count = convolution->feature_group_count(); in HandleConvolution()
Dhlo_cost_analysis.cc527 const int64 fma_count = (input_feature / convolution->feature_group_count()) * in HandleConvolution()
Dhlo_instructions.cc1856 int64 feature_group_count, int64 batch_group_count, const Window& window, in HloConvolutionInstruction() argument
1860 feature_group_count_(feature_group_count), in HloConvolutionInstruction()
1927 if (feature_group_count_ != other.feature_group_count()) { in IdenticalSlowPath()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h385 int64 feature_group_count = 1, int64 batch_group_count = 1,
392 int64 feature_group_count = 1, int64 batch_group_count = 1,
399 int64 feature_group_count = 1, int64 batch_group_count = 1,
406 int64 feature_group_count = 1, int64 batch_group_count = 1,
415 int64 feature_group_count = 1,
783 int64 feature_group_count, int64 batch_group_count,
789 int64 feature_group_count, int64 batch_group_count,
795 int64 feature_group_count, int64 batch_group_count,
801 int64 feature_group_count, int64 batch_group_count,
810 int64 feature_group_count, int64 batch_group_count,
[all …]
Dxla_builder.cc1101 int64 feature_group_count, int64 batch_group_count, in Conv() argument
1106 feature_group_count, batch_group_count, precision_config); in Conv()
1112 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralPadding() argument
1116 feature_group_count, batch_group_count, precision_config); in ConvWithGeneralPadding()
1122 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralDimensions() argument
1150 dimension_numbers, feature_group_count, in ConvWithGeneralDimensions()
1159 int64 feature_group_count, int64 batch_group_count, in ConvGeneral() argument
1162 dimension_numbers, feature_group_count, in ConvGeneral()
1171 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilated() argument
1193 lhs_shape, rhs_shape, feature_group_count, in ConvGeneralDilated()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py235 feature_group_count=1, argument
269 feature_group_count=feature_group_count,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1620 def Conv(self, lhs, rhs, window_strides, padding, feature_group_count=1): argument
1637 feature_group_count=feature_group_count)
1640 lhs_dilation, rhs_dilation, feature_group_count=1): argument
1657 dimension_numbers=None, feature_group_count=feature_group_count)
1676 feature_group_count=1): argument
1731 dimension_numbers, feature_group_count)
Dlocal_computation_builder.cc592 int64 feature_group_count) { in ConvGeneralDilated() argument
595 feature_group_count); in ConvGeneralDilated()
Dlocal_computation_builder.h300 int64 feature_group_count);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc134 new_conv_shape, new_input, new_kernel, hlo->feature_group_count(), in Run()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md619 | `feature_group_count` | int64 | the number of feature |
640 * `input-z`: The size of this dimension times `feature_group_count` should
674 The `feature_group_count` argument (default value 1) can be used for grouped
675 convolutions. `feature_group_count` needs to be a divisor of both the input and
676 the output feature dimension. If `feature_group_count` is greater than 1, it
678 output feature dimension are split evenly into `feature_group_count` many
681 dimension divided by `feature_group_count` (so it already has the size of a
683 `feature_group_count` many separate convolutions. The results of these
686 For depthwise convolution the `feature_group_count` argument would be set to the

12