| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bias_ops.cc | 53 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
57 ctx, feature_dim >= 0, in Compile()
61 ctx, bias_shape.dim_size(0) == input_shape.dim_size(feature_dim), in Compile()
67 xla::XlaOp result = xla::Add(ctx->Input(0), ctx->Input(1), {feature_dim}); in Compile()
98 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile() local
102 ctx, feature_dim >= 0, in Compile()
107 std::iota(reduce_dims.begin(), reduce_dims.begin() + feature_dim, 0); in Compile()
108 std::iota(reduce_dims.begin() + feature_dim, reduce_dims.end(), in Compile()
109 feature_dim + 1); in Compile()
|
| D | depthtospace_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
88 reshaped_shape.push_back(input_shape[feature_dim] / block_elems); in Compile()
95 transpose_order.push_back(feature_dim + num_spatial_dims); in Compile()
101 output_shape.push_back(input_shape[feature_dim] / block_elems); in Compile()
110 reshaped_shape.push_back(input_shape[feature_dim] / block_elems); in Compile()
123 output_shape.push_back(input_shape[feature_dim] / block_elems); in Compile()
140 input_shape[feature_dim] % (block_size_ * block_size_) == 0, in Compile()
|
| D | spacetodepth_op.cc | 69 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format); in Compile() local
93 reshaped_shape.push_back(input_shape[feature_dim]); in Compile()
102 transpose_order.push_back(feature_dim + num_spatial_dims); in Compile()
108 output_shape.push_back(input_shape[feature_dim] * block_elems); in Compile()
121 reshaped_shape.push_back(input_shape[feature_dim]); in Compile()
131 transpose_order.push_back(feature_dim); in Compile()
137 output_shape.push_back(input_shape[feature_dim] * block_elems); in Compile()
|
| D | extract_image_patches_op.cc | 62 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile() local
64 ctx, ksizes_[batch_dim] == 1 && ksizes_[feature_dim] == 1, in Compile()
69 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1, in Compile()
73 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1, in Compile()
102 const int64 depth = input_shape.dim_size(feature_dim); in Compile()
134 dims.set_input_feature_dimension(feature_dim); in Compile()
135 dims.set_output_feature_dimension(feature_dim); in Compile()
|
| D | conv_op_helpers.cc | 107 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
108 if (attrs.strides[batch_dim] != 1 || attrs.strides[feature_dim] != 1) { in CheckConvAttrs()
118 if (attrs.dilations[batch_dim] != 1 || attrs.dilations[feature_dim] != 1) { in CheckConvAttrs()
211 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
215 in_depth = input_shape.dimensions(feature_dim); in MakeXlaForwardConvOp()
242 dims.set_input_feature_dimension(feature_dim); in MakeXlaForwardConvOp()
243 dims.set_output_feature_dimension(feature_dim); in MakeXlaForwardConvOp()
301 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
308 int64 in_depth = input_shape.dimensions(feature_dim), in MakeXlaBackpropInputConvOp()
330 dnums.set_input_feature_dimension(feature_dim); in MakeXlaBackpropInputConvOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 38 int64_t feature_dim, in BroadcastToFeatureDim() argument
42 auto dims = DenseIntElementsAttr::get(dims_type, {feature_dim}); in BroadcastToFeatureDim()
126 int64_t feature_dim = bn_op.feature_index(); in matchAndRewrite() local
148 shape_value, feature_dim, rewriter); in matchAndRewrite()
151 shape_value, feature_dim, rewriter); in matchAndRewrite()
154 shape_value, feature_dim, rewriter); in matchAndRewrite()
156 bn_op.getLoc(), input_type, stddev, shape_value, feature_dim, rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1025 const int32 feature_dim = stats_summary_indices(idx, 1); in Compute() local
1029 FeatureMapIterator::value_type(feature_dim, BucketMap())); in Compute()
1138 const int32 feature_dim = f_iter->first; in process_node() local
1195 best_f_dim = feature_dim; in process_node()
1213 best_f_dim = feature_dim; in process_node()
1359 for (int feature_dim = 0; feature_dim < feature_dims; ++feature_dim) { in Compute() local
1360 const int32 feature_value = feature(i, feature_dim); in Compute()
1364 temp_stats_double(node, feature_dim, bucket, stat_dim) += in Compute()
1368 temp_stats_double(node, feature_dim, bucket, stat_dim) += in Compute()
1393 StatsPartitionKey(const int32 node_id, const int32 feature_dim, in StatsPartitionKey()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.cc | 124 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
125 dims->in_depth = input_shape.dim_size(feature_dim); in ConvBackpropComputeDimensionsV2()
135 if (dims->out_depth != out_backprop_shape.dim_size(feature_dim)) { in ConvBackpropComputeDimensionsV2()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 506 Value broadcast_from, int64_t feature_dim, in Broadcast1DToFeatureDim() argument
508 auto broadcast_dims = GetI64ElementsAttr({feature_dim}, &builder); in Broadcast1DToFeatureDim()
1098 IntegerAttr feature_dim = in GetConvDimensionNumbersAttr() local
1115 batch_dim, feature_dim, spatial_dims, kernel_input_feature_dim, in GetConvDimensionNumbersAttr()
1117 feature_dim, spatial_dims, builder->getContext())); in GetConvDimensionNumbersAttr()
1132 auto feature_dim = GetFeatureDimension( in matchAndRewrite() local
1135 feature_dim, rewriter); in matchAndRewrite()
1831 auto feature_dim = feature_dim_attr.getValue().getSExtValue(); in matchAndRewrite() local
1839 {GetDimSize(act_type, feature_dim)}, kernel_type); in matchAndRewrite()
1845 feature_dim); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 104 int64_t feature_dim = GetTensorFeatureDimIndex(rank, format); in GetBiasAddGradReductionIndices() local
106 std::iota(dims_to_reduce.begin(), dims_to_reduce.begin() + feature_dim, 0); in GetBiasAddGradReductionIndices()
107 std::iota(dims_to_reduce.begin() + feature_dim, dims_to_reduce.end(), in GetBiasAddGradReductionIndices()
108 feature_dim + 1); in GetBiasAddGradReductionIndices()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 385 DimensionHandle feature_dim = c->Dim(c->input(3), 1); in __anon770f674c0a02() local
388 c->set_output(0, c->MakeShape({max_splits, feature_dim, in __anon770f674c0a02()
|
| /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
480 if (feature_dim != -1 && bias_len != -1 && feature_dim != bias_len) { in Verify()
484 << feature_dim << " and " << bias_len << ", respectively"; in Verify()
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1619 feature_dim = 2
1626 max_splits, num_buckets, batch_size, feature_dim, logits_dims,
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5936 feature_dim = kernel_shape[1]
5956 xs.append(reshape(inputs[slices], (1, -1, feature_dim)))
|