| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 38 int64_t feature_dim, in BroadcastToFeatureDim()
127 int64_t feature_dim = bn_op.feature_index(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bias_ops.cc | 54 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
|
| D | conv_op_helpers.cc | 107 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
210 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
301 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
|
| D | depthtospace_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
|
| D | extract_image_patches_op.cc | 62 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile() local
|
| D | spacetodepth_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 125 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1068 const int32_t feature_dim = stats_summary_indices(idx, 1); in Compute() local
1188 const int32_t feature_dim = f_iter->first; in process_node() local
1409 for (int feature_dim = 0; feature_dim < feature_dims; ++feature_dim) { in Compute() local
1473 int32 feature_dim; member
1510 int32 feature_dim; member
1515 const int32_t instance, const int32_t feature_dim, const int32_t bucket_id, in AddInstanceStatsToMap()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 179 bool NeedsReformatTypeAndPermutation(int batch_dim, int feature_dim, in NeedsReformatTypeAndPermutation()
192 GetReformatTypeAndPermutation(int batch_dim, int feature_dim, in GetReformatTypeAndPermutation()
218 Value FormatToNHWC(Value value, int batch_dim, int feature_dim, in FormatToNHWC()
|
| D | lower_tf.cc | 150 int64_t feature_dim = GetTensorFeatureDimIndex(rank, format); in GetBiasAddGradReductionIndices() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 385 DimensionHandle feature_dim = c->Dim(c->input(3), 1); in __anon16fb5b110a02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 456 Value broadcast_from, int64_t feature_dim, in Broadcast1DToFeatureDim()
1082 IntegerAttr feature_dim = in GetConvDimensionNumbersAttr() local
1120 auto feature_dim = GetFeatureDimension( in matchAndRewrite() local
2441 auto feature_dim = feature_dim_attr.getValue().getSExtValue(); in matchAndRewrite() local
2553 auto feature_dim = getFeatureDimensionAttr(rewriter, data_format, op.x()); in matchAndRewrite() local
5093 const int feature_dim = in matchAndRewrite() local
5258 const int feature_dim = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 478 int64_t feature_dim = value_ty.getDimSize(feature_dim_idx); in Verify() local
|