| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 205 int64 batch_group_count = convolution->batch_group_count(); in HandleBatchGroupCount() local
207 if (batch_group_count == 1) { in HandleBatchGroupCount()
211 VLOG(2) << "Dealing with batch_group_count " << batch_group_count in HandleBatchGroupCount()
225 TF_RET_CHECK(input_batch == batch_group_count); in HandleBatchGroupCount()
236 output_feature_dimension, batch_group_count, add); in HandleBatchGroupCount()
238 convolution->shape(), batch_group_count, output_batch_dimension); in HandleBatchGroupCount()
293 dim->set_stride(batch_group_count); in HandleBatchGroupCount()
294 dim->set_size(batch_group_count); in HandleBatchGroupCount()
|
| D | shape_inference.cc | 1573 int64 batch_group_count, const Window& window, in InferConvolveShape() argument
1584 if (batch_group_count <= 0) { in InferConvolveShape()
1587 batch_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()
1710 if (batch_group_count > 1 && input_batch % kernel_output_features != 0) { in InferConvolveShape()
1716 input_batch, kernel_output_features, batch_group_count, in InferConvolveShape()
1751 if (input_batch % batch_group_count > 0) { in InferConvolveShape()
1757 input_batch, batch_group_count, ShapeUtil::HumanString(lhs), in InferConvolveShape()
1783 dimensions[dnums.output_batch_dimension()] = input_batch / batch_group_count; in InferConvolveShape()
|
| D | hlo_instructions.h | 1033 int64 feature_group_count, int64 batch_group_count, const Window& window,
1051 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count() function
1206 void set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() argument
1207 batch_group_count_ = batch_group_count; in set_batch_group_count()
1210 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count() function
|
| D | transpose_folding.cc | 179 convolution.batch_group_count(), convolution.window(), new_dnums, in FoldTransposeIntoConvolution()
|
| D | hlo_instruction.cc | 424 std::max<int64>(proto.batch_group_count(), 1), proto.window(), in CreateFromProto()
475 std::max(static_cast<int64>(proto.batch_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()
3472 int64 HloInstruction::batch_group_count() const { in batch_group_count() function in xla::HloInstruction
3474 return convolution->batch_group_count(); in batch_group_count()
3476 return Cast<HloCustomCallInstruction>(this)->batch_group_count(); in batch_group_count()
3479 void HloInstruction::set_batch_group_count(int64 batch_group_count) { in set_batch_group_count() argument
3481 batch_group_count); in set_batch_group_count()
|
| D | hlo_instruction.h | 439 int64 feature_group_count, int64 batch_group_count, const Window& window,
1564 int64 batch_group_count() const;
1566 void set_batch_group_count(int64 batch_group_count);
|
| D | shape_inference.h | 112 int64 batch_group_count, const Window& window,
|
| D | hlo_parser.cc | 1068 optional<int64> batch_group_count; in ParseInstructionRhs() local
1075 &batch_group_count}; in ParseInstructionRhs()
1089 if (!batch_group_count) { in ParseInstructionRhs()
1090 batch_group_count = 1; in ParseInstructionRhs()
1102 feature_group_count.value(), batch_group_count.value(), *window, in ParseInstructionRhs()
1489 optional<int64> batch_group_count; in ParseInstructionRhs() local
1500 &batch_group_count}; in ParseInstructionRhs()
1556 if (batch_group_count.has_value()) { in ParseInstructionRhs()
1557 instruction->set_batch_group_count(*batch_group_count); in ParseInstructionRhs()
|
| D | hlo_evaluator_typed_visitor.h | 1040 conv->batch_group_count(), window, dnums)); in HandleConvolution()
1064 const int64 batch_group_count = conv->batch_group_count(); in HandleConvolution() local
1069 batch_group_count](const absl::Span<const int64> out_index) { in HandleConvolution()
1086 const int64 batch_group_size = input_batch_size / batch_group_count; in HandleConvolution()
|
| D | hlo_cost_analysis.cc | 529 (batch / convolution->batch_group_count()) * in HandleConvolution()
|
| D | hlo_instructions.cc | 1856 int64 feature_group_count, int64 batch_group_count, const Window& window, in HloConvolutionInstruction() argument
1861 batch_group_count_(batch_group_count), in HloConvolutionInstruction()
1930 if (batch_group_count_ != other.batch_group_count()) { in IdenticalSlowPath()
|
| D | hlo.proto | 86 int64 batch_group_count = 58; field
|
| D | hlo_verifier.cc | 159 convolution->feature_group_count(), convolution->batch_group_count(), in HandleConvolution()
|
| /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,
416 int64 batch_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
1151 batch_group_count, precision_config); in ConvWithGeneralDimensions()
1159 int64 feature_group_count, int64 batch_group_count, in ConvGeneral() argument
1163 batch_group_count, precision_config); in ConvGeneral()
1171 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilated() argument
1194 batch_group_count, instr.window(), dimension_numbers)); in ConvGeneralDilated()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization.cc | 135 hlo->batch_group_count(), hlo->window(), new_dnums, in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 621 | `batch_group_count` | int64 | the number of batch |
692 The `batch_group_count` (default value 1) argument can be used for depthwise
693 filters during backpropagation. `batch_group_count` needs to be a divisor of the
694 size of the `lhs` (input) batch dimension. If `batch_group_count` is greater
696 `batch_group_size` where `batch_group_size = input batch / batch_group_count`.
697 For convolutions with `batch_group_count` greater than 1, the input batch size
699 that the output feature size must be equal to batch_group_count. Conceptually,
706 * `batch`: The size of this dimension times `batch_group_count` should equal
|