| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_conv2d.cc | 31 int64_t rhs_col_dilation, int64_t feature_group_count) { in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
51 int64_t feature_group_count) { in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
|
| D | runtime_conv2d.cc | 34 int64_t feature_group_count) { in __xla_cpu_runtime_EigenConv2DF32()
56 int64_t rhs_col_dilation, int64_t feature_group_count) { in __xla_cpu_runtime_EigenConv2DF16()
|
| D | runtime_conv3d.cc | 36 int64_t feature_group_count) { in __xla_cpu_runtime_EigenConv3DF32()
61 int64_t feature_group_count) { in __xla_cpu_runtime_EigenConv3DF16()
|
| D | runtime_single_threaded_conv3d.cc | 33 int64_t feature_group_count) { in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
56 int64_t feature_group_count) { in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
|
| D | runtime_conv_impl.h | 40 Eigen::Index rhs_y_dilation, Eigen::Index feature_group_count) { in EigenConv2DImpl()
121 Eigen::Index feature_group_count) { in EigenConv3DImpl()
|
| D | runtime_conv2d_acl.cc | 38 int64_t feature_group_count) { in ACLDepthwiseConvImpl()
242 int64_t feature_group_count) { in __xla_cpu_runtime_ACLConv2DF32()
|
| D | runtime_conv2d_acl.h | 86 int64_t rhs_col_dilation, int64_t feature_group_count) { in __xla_cpu_runtime_ACLConv2DF32()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 58 Value &rhs_dilation, Value &feature_group_count) { in PrepareXlaConvParams()
131 feature_group_count; in CreateXLAConvOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| 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/client/ |
| D | xla_builder.cc | 1447 Padding padding, int64_t feature_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 …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 196 int64_t feature_group_count; member
|
| D | gpu_conv_rewriter.cc | 564 int64_t feature_group_count, in CreateGpuConv()
|
| D | jitrt_custom_calls.cc | 698 int64_t feature_group_count; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 1611 int64_t feature_group_count() const { return feature_group_count_; } in feature_group_count() function
1849 void set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count()
1860 int64_t feature_group_count() const { return feature_group_count_; } in feature_group_count() function
|
| D | hlo.proto | 145 int64 feature_group_count = 50; field
|
| D | hlo_creation_utils.cc | 118 HloInstruction* lhs, HloInstruction* rhs, int64_t feature_group_count, in MakeConvolveHlo()
|
| D | hlo_instruction.cc | 1172 int64_t feature_group_count, int64_t batch_group_count, in CreateConvolve()
4739 int64_t HloInstruction::feature_group_count() const { in feature_group_count() function in xla::HloInstruction
4746 void HloInstruction::set_feature_group_count(int64_t feature_group_count) { in set_feature_group_count()
|
| D | hlo_parser.cc | 1985 optional<int64_t> feature_group_count; in CreateInstruction() local
2586 optional<int64_t> feature_group_count; in CreateInstruction() local
|
| D | shape_inference.cc | 1615 const Shape& lhs, const Shape& rhs, int64_t feature_group_count, in InferConvolveShape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 250 feature_group_count=1, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 937 int64_t feature_group_count = conv.feature_group_count(); in CreateShardedConvConvolution() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 217 const int64_t feature_group_count = conv_op.feature_group_count(); in matchAndRewrite() local
431 int feature_group_count = conv_op.feature_group_count(); in matchAndRewrite() local
819 int feature_group_count = conv_op.feature_group_count(); in IsSupportedConvOp() local
|
| D | shape_inference.cc | 1601 auto feature_group_count = op.feature_group_count(); in PrecheckForXlaConvV2Op() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 107 int64_t feature_group_count, int64_t batch_group_count, in ConvGeneralDilatedInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1215 const int64_t feature_group_count = in matchAndRewriteDynamicConv() local
1371 const int64_t feature_group_count = in matchAndRewrite() local
5003 const int64_t feature_group_count = in_depth / filter_in_depth; in matchAndRewrite() local
|