| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 52 int64_t feature_group_count; in Compile() local
|
| D | conv_op_helpers.cc | 222 int64_t feature_group_count = in_depth / filter_in_depth; in MakeXlaForwardConvOp() local
310 feature_group_count = in MakeXlaBackpropInputConvOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 1311 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count() function
1531 void set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count()
1542 int64 feature_group_count() const { return feature_group_count_; } in feature_group_count() function
|
| D | hlo.proto | 125 int64 feature_group_count = 50; field
|
| D | hlo_creation_utils.cc | 94 HloInstruction* lhs, HloInstruction* rhs, int64_t feature_group_count, in MakeConvolveHlo()
|
| D | hlo_instruction.cc | 1102 int64_t feature_group_count, int64_t batch_group_count, in CreateConvolve()
4464 int64 HloInstruction::feature_group_count() const { in feature_group_count() function in xla::HloInstruction
4471 void HloInstruction::set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count()
|
| D | hlo_parser.cc | 1746 optional<int64> feature_group_count; in ParseInstructionRhs() local
2377 optional<int64> feature_group_count; in ParseInstructionRhs() local
|
| D | shape_inference.cc | 1605 const Shape& lhs, const Shape& rhs, int64_t feature_group_count, in InferConvolveShape()
|
| D | hlo_instructions.cc | 2273 int64_t feature_group_count, int64_t batch_group_count, in HloConvolutionInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 134 int64 feature_group_count; member
|
| D | gpu_conv_rewriter.cc | 44 int64_t feature_group_count, in CreateGpuConv()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1387 int64_t feature_group_count, int64_t batch_group_count, in Conv()
1400 int64_t feature_group_count, int64_t batch_group_count, in ConvWithGeneralPadding()
1412 int64_t feature_group_count, int64_t batch_group_count, in ConvWithGeneralDimensions()
1451 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneral()
1465 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilated()
1503 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvInstruction()
1545 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvInputGrad()
1568 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvKernelGrad()
1594 int64_t feature_group_count, int64_t batch_group_count, in DynamicConvForward()
1616 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilatedInternal()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 253 feature_group_count=1, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 104 int feature_group_count = conv_op.feature_group_count(); in matchAndRewrite() local
478 int feature_group_count = conv_op.feature_group_count(); in IsSupportedConvOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 99 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilatedInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 935 int64_t feature_group_count = conv.feature_group_count(); in CreateShardedConvConvolution() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1395 const int64_t feature_group_count = in matchAndRewriteDynamicConv() local
1546 const int64_t feature_group_count = in matchAndRewrite() local
5097 const int64_t feature_group_count = in_depth / filter_in_depth; in matchAndRewrite() local
|