Home
last modified time | relevance | path

Searched refs:feature_dim (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc53 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()
Ddepthtospace_op.cc69 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()
Dspacetodepth_op.cc69 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()
Dextract_image_patches_op.cc55 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile() local
57 ctx, ksizes_[batch_dim] == 1 && ksizes_[feature_dim] == 1, in Compile()
62 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1, in Compile()
66 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1, in Compile()
95 const int64 depth = input_shape.dim_size(feature_dim); in Compile()
128 dims.set_input_feature_dimension(feature_dim); in Compile()
129 dims.set_output_feature_dimension(feature_dim); in Compile()
Dconv_op_helpers.cc174 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in CheckConvAttrs() local
175 if (attrs.strides[batch_dim] != 1 || attrs.strides[feature_dim] != 1) { in CheckConvAttrs()
184 if (attrs.dilations[batch_dim] != 1 || attrs.dilations[feature_dim] != 1) { in CheckConvAttrs()
270 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaForwardConvOp() local
275 if (in_depth != input_shape.dimensions(feature_dim)) { in MakeXlaForwardConvOp()
278 input_shape.dimensions(feature_dim)); in MakeXlaForwardConvOp()
293 dims.set_input_feature_dimension(feature_dim); in MakeXlaForwardConvOp()
294 dims.set_output_feature_dimension(feature_dim); in MakeXlaForwardConvOp()
330 int feature_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropInputConvOp() local
354 dnums.set_input_feature_dimension(feature_dim); in MakeXlaBackpropInputConvOp()
[all …]
/external/tensorflow/tensorflow/contrib/image/python/kernel_tests/
Dinterpolate_spline_test.py239 feature_dim = query_points.shape[-1]
242 dtype=train_points.dtype, shape=[None, None, feature_dim])
246 dtype=query_points.dtype, shape=[None, None, feature_dim])
279 feature_dim = query_points.shape[-1]
282 dtype=train_points.dtype, shape=[None, None, feature_dim])
290 dtype=query_points.dtype, shape=[None, None, feature_dim])
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_ops.cc128 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2() local
129 dims->in_depth = input_shape.dim_size(feature_dim); in ConvBackpropComputeDimensionsV2()
139 if (dims->out_depth != out_backprop_shape.dim_size(feature_dim)) { in ConvBackpropComputeDimensionsV2()
/external/tensorflow/tensorflow/python/keras/
Dbackend.py4807 feature_dim = kernel_shape[1]
4827 xs.append(reshape(inputs[slices], (1, -1, feature_dim)))