| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bias_ops.cc | 50 int feature_dim = (data_format_ == FORMAT_NHWC) ? input_shape.dims() - 1 variable
53 ctx, feature_dim >= 0,
57 ctx, bias_shape.dim_size(0) == input_shape.dim_size(feature_dim),
64 ctx->builder()->Add(ctx->Input(0), ctx->Input(1), {feature_dim});
94 int feature_dim = (data_format_ == FORMAT_NHWC) variable
98 ctx, feature_dim >= 0,
103 std::iota(reduce_dims.begin(), reduce_dims.begin() + feature_dim, 0);
104 std::iota(reduce_dims.begin() + feature_dim, reduce_dims.end(),
105 feature_dim + 1);
|
| depthtospace_op.cc | 56 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format_); variable
75 reshaped_shape.push_back(input_shape[feature_dim] / block_elems);
82 transpose_order.push_back(feature_dim + num_spatial_dims);
88 output_shape.push_back(input_shape[feature_dim] / block_elems);
97 reshaped_shape.push_back(input_shape[feature_dim] / block_elems);
110 output_shape.push_back(input_shape[feature_dim] / block_elems);
127 input_shape[feature_dim] % (block_size_ * block_size_) == 0,
|
| spacetodepth_op.cc | 56 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format_); variable
80 reshaped_shape.push_back(input_shape[feature_dim]);
89 transpose_order.push_back(feature_dim + num_spatial_dims);
95 output_shape.push_back(input_shape[feature_dim] * block_elems);
108 reshaped_shape.push_back(input_shape[feature_dim]);
118 transpose_order.push_back(feature_dim);
124 output_shape.push_back(input_shape[feature_dim] * block_elems);
|
| extract_image_patches_op.cc | 54 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); variable
56 ctx, ksizes_[batch_dim] == 1 && ksizes_[feature_dim] == 1,
61 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1,
65 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1,
94 const int64 depth = input_shape.dim_size(feature_dim);
129 dims.set_input_feature_dimension(feature_dim);
130 dims.set_output_feature_dimension(feature_dim);
|
| conv_ops.cc | 200 int feature_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); variable
202 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1,
211 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1,
242 OP_REQUIRES(ctx, in_depth == input_shape.dim_size(feature_dim),
245 " vs ", input_shape.dim_size(feature_dim)));
266 dims.set_input_feature_dimension(feature_dim);
267 dims.set_output_feature_dimension(feature_dim);
351 int feature_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); variable
353 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1,
362 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_ops.cc | 126 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); local
127 dims->in_depth = input_shape.dim_size(feature_dim);
135 if (dims->out_depth != out_backprop_shape.dim_size(feature_dim)) {
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| backend.py | [all...] |
