Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h1051 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()
1210 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count() function
Dconvolution_group_converter.cc205 int64 batch_group_count = convolution->batch_group_count(); in HandleBatchGroupCount() local
Dhlo.proto86 int64 batch_group_count = 58; field
Dhlo_instruction.cc782 int64 feature_group_count, int64 batch_group_count, const Window& window, in CreateConvolve()
3472 int64 HloInstruction::batch_group_count() const { in batch_group_count() function in xla::HloInstruction
3479 void HloInstruction::set_batch_group_count(int64 batch_group_count) { in set_batch_group_count()
Dhlo_parser.cc1068 optional<int64> batch_group_count; in ParseInstructionRhs() local
1489 optional<int64> batch_group_count; in ParseInstructionRhs() local
Dshape_inference.cc1573 int64 batch_group_count, const Window& window, in InferConvolveShape()
Dhlo_evaluator_typed_visitor.h1064 const int64 batch_group_count = conv->batch_group_count(); in HandleConvolution() local
Dhlo_instructions.cc1856 int64 feature_group_count, int64 batch_group_count, const Window& window, in HloConvolutionInstruction()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1101 int64 feature_group_count, int64 batch_group_count, in Conv()
1112 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralPadding()
1122 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralDimensions()
1159 int64 feature_group_count, int64 batch_group_count, in ConvGeneral()
1171 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilated()
2972 int64 feature_group_count, int64 batch_group_count, in Conv()
2982 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralPadding()
2992 int64 feature_group_count, int64 batch_group_count, in ConvWithGeneralDimensions()
3003 int64 feature_group_count, int64 batch_group_count, in ConvGeneral()
3016 int64 feature_group_count, int64 batch_group_count, in ConvGeneralDilated()