| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 1448 int64_t batch_group_count, in Conv()
1461 int64_t feature_group_count, int64_t batch_group_count, in ConvWithGeneralPadding()
1473 int64_t feature_group_count, int64_t batch_group_count, in ConvWithGeneralDimensions()
1512 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneral()
1530 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilated()
1570 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvInstruction()
1614 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvInputGrad()
1639 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvKernelGrad()
1666 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvForward()
1689 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilatedInternal()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instructions.h | 1616 int64_t batch_group_count() const { return batch_group_count_; } in batch_group_count() function
1852 void set_batch_group_count(int64_t batch_group_count) { in set_batch_group_count()
1861 int64_t batch_group_count() const { return batch_group_count_; } in batch_group_count() function
|
| H A D | convolution_group_converter.cc | 208 int64_t batch_group_count = convolution->batch_group_count(); in HandleBatchGroupCount() local
|
| H A D | hlo.proto | 147 int64 batch_group_count = 58; field
|
| H A D | hlo_instruction.cc | 1172 int64_t feature_group_count, int64_t batch_group_count, in CreateConvolve()
4754 int64_t HloInstruction::batch_group_count() const { in batch_group_count() function in xla::HloInstruction
4761 void HloInstruction::set_batch_group_count(int64_t batch_group_count) { in set_batch_group_count()
|
| H A D | hlo_creation_utils.cc | 119 int64_t batch_group_count, const Window& window, in MakeConvolveHlo()
|
| H A D | hlo_parser.cc | 1986 optional<int64_t> batch_group_count; in CreateInstruction() local
2587 optional<int64_t> batch_group_count; in CreateInstruction() local
|
| H A D | shape_inference.cc | 1616 int64_t batch_group_count, const Window& window, in InferConvolveShape()
|
| H A D | hlo_instructions.cc | 2433 int64_t feature_group_count, int64_t batch_group_count, in HloConvolutionInstruction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 1544 llvm::SmallVector<int64_t> rhs_dilations, int64_t batch_group_count, in InferXlaConvOutputShape()
1602 int64_t batch_group_count = op.batch_group_count(); in PrecheckForXlaConvV2Op() local
1734 int64_t batch_group_count = op.batch_group_count(); in InferShapeForXlaConvV2Op() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | conv_op_helpers.cc | 434 batch_group_count = in MakeXlaBackpropFilterConvOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 255 batch_group_count=1): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | convolution_handler.cc | 945 int64_t batch_group_count = conv.batch_group_count(); in CreateShardedConvConvolution() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 107 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilatedInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 5165 const int64_t batch_group_count = in_depth / filter_in_depth; in matchAndRewrite() local
|