Home
last modified time | relevance | path

Searched defs:batch_group_count (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h1247 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count() function
1449 void set_batch_group_count(int64 batch_group_count) { in set_batch_group_count()
1458 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count() function
Dconvolution_group_converter.cc204 int64 batch_group_count = convolution->batch_group_count(); in HandleBatchGroupCount() local
Dhlo.proto92 int64 batch_group_count = 58; field
Dhlo_creation_utils.cc95 int64 batch_group_count, const Window& window, in MakeConvolveHlo()
Dhlo_instruction.cc985 int64 feature_group_count, int64 batch_group_count, const Window& window, in CreateConvolve()
4181 int64 HloInstruction::batch_group_count() const { in batch_group_count() function in xla::HloInstruction
4188 void HloInstruction::set_batch_group_count(int64 batch_group_count) { in set_batch_group_count()
Dhlo_parser.cc1610 optional<int64> batch_group_count; in ParseInstructionRhs() local
2241 optional<int64> batch_group_count; in ParseInstructionRhs() local
Dshape_inference.cc1617 int64 batch_group_count, const Window& window, in InferConvolveShape()
Dhlo_instructions.cc2135 int64 feature_group_count, int64 batch_group_count, const Window& window, in HloConvolutionInstruction()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1375 int64 feature_group_count, int64 batch_group_count, in Conv()
1388 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralPadding()
1400 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralDimensions()
1439 int64 feature_group_count, int64 batch_group_count, in ConvGeneral()
1453 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilated()
1491 int64 feature_group_count, int64 batch_group_count, in DynamicConvInstruction()
1533 int64 feature_group_count, int64 batch_group_count, in DynamicConvInputGrad()
1556 int64 feature_group_count, int64 batch_group_count, in DynamicConvKernelGrad()
1582 int64 feature_group_count, int64 batch_group_count, in DynamicConvForward()
1604 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilatedInternal()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc434 batch_group_count = in MakeXlaBackpropFilterConvOp() local
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc99 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilatedInternal()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc943 int64 batch_group_count = conv.batch_group_count(); in CreateShardedConvConvolution() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4358 const int64_t batch_group_count = in_depth / filter_in_depth; in matchAndRewrite() local