Searched defs:feature_dim (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bias_ops.cc | 53 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
98 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
211 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/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 38 int64_t feature_dim, in BroadcastToFeatureDim()
126 int64_t feature_dim = bn_op.feature_index(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 124 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1025 const int32 feature_dim = stats_summary_indices(idx, 1); in Compute() local
1138 const int32 feature_dim = f_iter->first; in process_node() local
1359 for (int feature_dim = 0; feature_dim < feature_dims; ++feature_dim) { in Compute() local
1423 int32 feature_dim; member
1460 int32 feature_dim; member
1464 static void AddInstanceStatsToMap(const int32 instance, const int32 feature_dim, in AddInstanceStatsToMap()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 385 DimensionHandle feature_dim = c->Dim(c->input(3), 1); in __anon770f674c0a02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 506 Value broadcast_from, int64_t feature_dim, in Broadcast1DToFeatureDim()
1098 IntegerAttr feature_dim = in GetConvDimensionNumbersAttr() local
1132 auto feature_dim = GetFeatureDimension( in matchAndRewrite() local
1831 auto feature_dim = feature_dim_attr.getValue().getSExtValue(); in matchAndRewrite() local
1939 auto feature_dim = in matchAndRewrite() local
4189 const int feature_dim = in matchAndRewrite() local
4354 const int feature_dim = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 104 int64_t feature_dim = GetTensorFeatureDimIndex(rank, format); in GetBiasAddGradReductionIndices() 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
|