Searched defs:batch_group_count (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_instructions.h | 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() 1210 int64 batch_group_count() const { return batch_group_count_; } in batch_group_count() function
|
D | convolution_group_converter.cc | 205 int64 batch_group_count = convolution->batch_group_count(); in HandleBatchGroupCount() local
|
D | hlo.proto | 86 int64 batch_group_count = 58; field
|
D | hlo_instruction.cc | 782 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()
|
D | hlo_parser.cc | 1068 optional<int64> batch_group_count; in ParseInstructionRhs() local 1489 optional<int64> batch_group_count; in ParseInstructionRhs() local
|
D | shape_inference.cc | 1573 int64 batch_group_count, const Window& window, in InferConvolveShape()
|
D | hlo_evaluator_typed_visitor.h | 1064 const int64 batch_group_count = conv->batch_group_count(); in HandleConvolution() local
|
D | hlo_instructions.cc | 1856 int64 feature_group_count, int64 batch_group_count, const Window& window, in HloConvolutionInstruction()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 1101 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()
|