Home
last modified time | relevance | path

Searched refs:input_dimensions (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dpadding.cc26 Status ValidatePaddingValues(absl::Span<const int64> input_dimensions, in ValidatePaddingValues() argument
29 bool ok = input_dimensions.size() == window_dimensions.size() && in ValidatePaddingValues()
30 input_dimensions.size() == window_strides.size(); in ValidatePaddingValues()
35 input_dimensions.size(), window_dimensions.size(), in ValidatePaddingValues()
42 absl::Span<const int64> input_dimensions, in MakePadding() argument
45 TF_CHECK_OK(ValidatePaddingValues(input_dimensions, window_dimensions, in MakePadding()
54 for (size_t i = 0; i < input_dimensions.size(); ++i) { in MakePadding()
55 int64 input_dimension = input_dimensions[i]; in MakePadding()
Dpadding.h44 Status ValidatePaddingValues(absl::Span<const int64> input_dimensions,
60 absl::Span<const int64> input_dimensions,
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
Dperiodic_resample_op.h45 auto input_dimensions = TensorShapeToVector(input_shape); in InputIndexer() local
49 ComputeDimensionCeiling(output_dimensions, input_dimensions); in InputIndexer()
60 last_index_factor *= input_dimensions[r]; in InputIndexer()
82 const std::vector<tensorflow::int64>& input_dimensions);
182 const std::vector<tensorflow::int64>& input_dimensions) { in ComputeDimensionCeiling() argument
183 std::vector<tensorflow::int64> dimension_ceiling(input_dimensions.size()); in ComputeDimensionCeiling()
184 for (size_t i = 0; i < input_dimensions.size(); ++i) { in ComputeDimensionCeiling()
185 dimension_ceiling[i] = (output_dimensions[i] + input_dimensions[i] - 1) / in ComputeDimensionCeiling()
186 input_dimensions[i]; in ComputeDimensionCeiling()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h392 const dnn::BatchDescriptor& input_dimensions,
399 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
409 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
428 const dnn::BatchDescriptor& input_dimensions,
436 const dnn::BatchDescriptor& input_dimensions,
444 const dnn::BatchDescriptor& input_dimensions,
452 const dnn::BatchDescriptor& input_dimensions,
460 const dnn::BatchDescriptor& input_dimensions,
470 const dnn::BatchDescriptor& input_dimensions,
480 const dnn::BatchDescriptor& input_dimensions,
[all …]
Dcuda_dnn.cc3528 const dnn::BatchDescriptor& input_dimensions, in DoMatMul() argument
3531 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
3544 if (input_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3545 input_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
3570 const int64 n = input_dimensions.count(); in DoMatMul()
3571 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3615 const uint64 n = input_dimensions.count(); in DoMatMul()
3616 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3626 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()
3730 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1500 const dnn::BatchDescriptor& input_dimensions,
1512 const dnn::BatchDescriptor& input_dimensions,
1524 const dnn::BatchDescriptor& input_dimensions,
1572 const dnn::BatchDescriptor& input_dimensions,
1580 const dnn::BatchDescriptor& input_dimensions,
1591 const dnn::BatchDescriptor& input_dimensions,
1602 const dnn::BatchDescriptor& input_dimensions,
1614 const dnn::BatchDescriptor& input_dimensions,
1627 const dnn::BatchDescriptor& input_dimensions,
1640 const dnn::BatchDescriptor& input_dimensions,
[all …]
Dstream.cc1141 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul() argument
1144 VLOG_CALL(PARAM(input_data), PARAM(weights), PARAM(input_dimensions), in ThenMatMul()
1149 CheckError(dnn->DoMatMul(this, input_data, weights, input_dimensions, in ThenMatMul()
1161 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized() argument
1165 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
1171 weight_scales, input_dimensions, in ThenMatMulQuantized()
1183 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized() argument
1187 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
1193 weight_scales, input_dimensions, in ThenMatMulQuantized()
1222 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward() argument
[all …]
Dstream.h494 const dnn::BatchDescriptor &input_dimensions,
501 const dnn::BatchDescriptor &input_dimensions,
508 const dnn::BatchDescriptor &input_dimensions,
518 const dnn::BatchDescriptor &input_dimensions,
525 const dnn::BatchDescriptor &input_dimensions,
532 const dnn::BatchDescriptor &input_dimensions,
539 const dnn::BatchDescriptor &input_dimensions,
546 const dnn::BatchDescriptor &input_dimensions,
555 const dnn::BatchDescriptor &input_dimensions,
564 const dnn::BatchDescriptor &input_dimensions,
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h383 const dnn::BatchDescriptor& input_dimensions,
390 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
400 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
419 const dnn::BatchDescriptor& input_dimensions,
427 const dnn::BatchDescriptor& input_dimensions,
435 const dnn::BatchDescriptor& input_dimensions,
443 const dnn::BatchDescriptor& input_dimensions,
453 const dnn::BatchDescriptor& input_dimensions,
463 const dnn::BatchDescriptor& input_dimensions,
487 Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions,
[all …]
Drocm_dnn.cc3288 const dnn::BatchDescriptor& input_dimensions, in DoMatMul() argument
3291 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
3304 if (input_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3305 input_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
3330 const int64 n = input_dimensions.count(); in DoMatMul()
3331 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3375 const uint64 n = input_dimensions.count(); in DoMatMul()
3376 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3386 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()
3477 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DREADME.md1379 ceil(input_dimensions[i] / strides[i]).
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2507 `∀ i . 0 ≤ i < rank ⇒ input_dimensions[permutation[i]] = output_dimensions[i]`.