Home
last modified time | relevance | path

Searched refs:spatial_dims (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc351 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size, in Compute()
352 dims.spatial_dims[0].stride, padding_, in Compute()
353 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); in Compute()
357 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size, in Compute()
358 dims.spatial_dims[1].stride, padding_, in Compute()
359 &dims.spatial_dims[1].output_size, &pad_left, &pad_right)); in Compute()
365 out_backprop.tensor<T, 4>(), dims.spatial_dims[0].input_size, in Compute()
366 dims.spatial_dims[1].input_size, in Compute()
367 static_cast<int>(dims.spatial_dims[0].stride), in Compute()
368 static_cast<int>(dims.spatial_dims[1].stride), in Compute()
[all …]
Dconv_grad_filter_ops.cc270 dims.spatial_dims[0].input_size, dims.spatial_dims[0].filter_size, in Compute()
271 dims.spatial_dims[0].stride, padding_, in Compute()
272 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); in Compute()
276 dims.spatial_dims[1].input_size, dims.spatial_dims[1].filter_size, in Compute()
277 dims.spatial_dims[1].stride, padding_, in Compute()
278 &dims.spatial_dims[1].output_size, &pad_left, &pad_right)); in Compute()
285 dims.spatial_dims[0].input_size, dims.spatial_dims[1].input_size, in Compute()
286 static_cast<int>(dims.spatial_dims[0].stride), in Compute()
287 static_cast<int>(dims.spatial_dims[1].stride), in Compute()
296 const int filter_total_size = dims.spatial_dims[0].filter_size * in Compute()
[all …]
Dconv_grad_ops_3d.cc247 static_cast<int>(dims.spatial_dims[0].stride), // stride_planes in Compute()
248 static_cast<int>(dims.spatial_dims[1].stride), // stride_rows in Compute()
249 static_cast<int>(dims.spatial_dims[2].stride)); // stride_cols in Compute()
353 dims.spatial_dims[0].input_size, in Compute()
354 dims.spatial_dims[0].filter_size, in Compute()
355 dims.spatial_dims[0].stride, padding_, in Compute()
356 &dims.spatial_dims[0].output_size, in Compute()
359 dims.spatial_dims[1].input_size, in Compute()
360 dims.spatial_dims[1].filter_size, in Compute()
361 dims.spatial_dims[1].stride, padding_, in Compute()
[all …]
Dconv_grad_ops.h236 gtl::InlinedVector<ConvBackpropSpatialDimension, 3> spatial_dims;
245 int64 input_size(int dim) const { return spatial_dims[dim].input_size; }
246 int64 filter_size(int dim) const { return spatial_dims[dim].filter_size; }
247 int64 output_size(int dim) const { return spatial_dims[dim].output_size; }
248 int64 stride(int dim) const { return spatial_dims[dim].stride; }
249 int64 dilation(int dim) const { return spatial_dims[dim].dilation; }
Dconv_grad_ops.cc143 dims->spatial_dims.resize(num_spatial_dims); in ConvBackpropComputeDimensionsV2()
154 &dims->spatial_dims[i])); in ConvBackpropComputeDimensionsV2()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py328 spatial_dims=None, argument
476 spatial_dims=spatial_dims,
505 spatial_dims=None, argument
525 if spatial_dims is None:
526 spatial_dims = range(starting_spatial_dim,
528 orig_spatial_dims = list(spatial_dims)
529 spatial_dims = sorted(set(int(x) for x in orig_spatial_dims))
530 if spatial_dims != orig_spatial_dims or any(x < 1 for x in spatial_dims):
536 expected_input_rank = spatial_dims[-1]
538 expected_input_rank = spatial_dims[-1] + 1
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc374 padding[i] = {dims.spatial_dims[i].pad_before, in MakeXlaBackpropInputConvOp()
375 dims.spatial_dims[i].pad_after}; in MakeXlaBackpropInputConvOp()
376 lhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropInputConvOp()
477 rhs_dilation[i] = dims.spatial_dims[i].stride; in MakeXlaBackpropFilterConvOp()
486 dims.spatial_dims[i].expanded_output_size + in MakeXlaBackpropFilterConvOp()
487 (dims.spatial_dims[i].filter_size - 1) * attrs.dilations[dim]; in MakeXlaBackpropFilterConvOp()
503 const int64 pad_total = padded_in_size - dims.spatial_dims[i].input_size; in MakeXlaBackpropFilterConvOp()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h589 std::vector<int64> spatial_dims(num_src_spatial_dims); in ShapeFromFormat()
591 spatial_dims[spatial_dim] = in ShapeFromFormat()
596 spatial_dims[num_src_spatial_dims - 1] *= 4; in ShapeFromFormat()
598 return ShapeFromFormat(dst_format, batch, {spatial_dims}, channels); in ShapeFromFormat()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc354 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape() argument
357 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in DimensionsFromShape()
361 for (int spatial_dim_index = 0; spatial_dim_index < spatial_dims.size(); in DimensionsFromShape()
363 spatial_dims[spatial_dim_index] = context->Dim( in DimensionsFromShape()
378 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions() argument
381 const int32 rank = GetTensorDimsFromSpatialDims(spatial_dims.size(), format); in ShapeFromDimensions()
387 for (int spatial_dim_index = 0; spatial_dim_index < spatial_dims.size(); in ShapeFromDimensions()
390 rank, format, spatial_dim_index)] = spatial_dims[spatial_dim_index]; in ShapeFromDimensions()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt329 …'input\', \'dilation_rate\', \'padding\', \'op\', \'filter_shape\', \'spatial_dims\', \'data_forma…
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc341 const std::vector<int64>& spatial_dims);
676 const std::vector<int64>& spatial_dims) { in ImageDims() argument
681 for (int dim : spatial_dims) { in ImageDims()
689 for (int dim : spatial_dims) { in ImageDims()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt421 …'input\', \'dilation_rate\', \'padding\', \'op\', \'filter_shape\', \'spatial_dims\', \'data_forma…
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc572 absl::Span<const int64> spatial_dims = in CudnnFilterDescriptor() local
574 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2); in CudnnFilterDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc614 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor() local
615 std::copy(spatial_dims.begin(), spatial_dims.end(), dims.begin() + 2); in ScopedFilterDescriptor()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md633 * `spatial_dims`: Describes the `n` spatial dimensions that define the base
642 * `spatial_dims`: Describes the `n` spatial dimensions that define the n-d
709 * `spatial_dims`: One value for each valid placement of the convolutional