/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_conv_rewriter.cc | 43 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()
|
D | cudnn_conv_runner.cc | 71 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/ |
D | xla_conv_op.cc | 50 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/ |
D | convolution_group_converter_test.cc | 58 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()
|
D | hlo_creation_utils.cc | 83 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()
|
D | shape_inference.cc | 1572 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()
|
D | hlo_creation_utils.h | 58 HloInstruction* lhs, HloInstruction* rhs, int64 feature_group_count,
|
D | hlo_instructions.h | 1033 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
|
D | transpose_folding.cc | 178 convolution.shape(), new_lhs, new_rhs, convolution.feature_group_count(), in FoldTransposeIntoConvolution()
|
D | hlo_instruction.cc | 423 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()
|
D | hlo_instruction.h | 439 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);
|
D | shape_inference.h | 111 const Shape& lhs, const Shape& rhs, int64 feature_group_count,
|
D | hlo_parser.cc | 1067 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()
|
D | hlo_evaluator_typed_visitor.h | 1039 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()
|
D | convolution_group_converter.cc | 328 int64 group_count = convolution->feature_group_count(); in HandleConvolution()
|
D | hlo_cost_analysis.cc | 527 const int64 fma_count = (input_feature / convolution->feature_group_count()) * in HandleConvolution()
|
D | hlo_instructions.cc | 1856 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/ |
D | xla_builder.h | 385 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 …]
|
D | xla_builder.cc | 1101 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/ |
D | xla.py | 235 feature_group_count=1, argument 269 feature_group_count=feature_group_count,
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xla_client.py | 1620 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)
|
D | local_computation_builder.cc | 592 int64 feature_group_count) { in ConvGeneralDilated() argument 595 feature_group_count); in ConvGeneralDilated()
|
D | local_computation_builder.h | 300 int64 feature_group_count);
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | conv_canonicalization.cc | 134 new_conv_shape, new_input, new_kernel, hlo->feature_group_count(), in Run()
|
/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
D | operation_semantics.md | 619 | `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
|