Home
last modified time | relevance | path

Searched refs:input_dimensions (Results 1 – 13 of 13) 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/stream_executor/rocm/
Drocm_dnn.h50 bool IsSame(const dnn::BatchDescriptor& input_dimensions,
61 bool find(const void* p, const dnn::BatchDescriptor& input_dimensions,
65 void insert(const void* p, const dnn::BatchDescriptor& input_dimensions,
435 const dnn::BatchDescriptor& input_dimensions,
442 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
452 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
471 const dnn::BatchDescriptor& input_dimensions,
479 const dnn::BatchDescriptor& input_dimensions,
487 const dnn::BatchDescriptor& input_dimensions,
495 const dnn::BatchDescriptor& input_dimensions,
[all …]
Drocm_dnn.cc3814 const dnn::BatchDescriptor& input_dimensions, in DoMatMul() argument
3817 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
3830 if (input_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
3831 input_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
3856 const int64 n = input_dimensions.count(); in DoMatMul()
3857 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3901 const uint64 n = input_dimensions.count(); in DoMatMul()
3902 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
3912 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()
4003 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h411 const dnn::BatchDescriptor& input_dimensions,
418 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
428 const dnn::BatchDescriptor& input_dimensions, in DoMatMulQuantized() argument
447 const dnn::BatchDescriptor& input_dimensions,
455 const dnn::BatchDescriptor& input_dimensions,
463 const dnn::BatchDescriptor& input_dimensions,
471 const dnn::BatchDescriptor& input_dimensions,
479 const dnn::BatchDescriptor& input_dimensions,
489 const dnn::BatchDescriptor& input_dimensions,
499 const dnn::BatchDescriptor& input_dimensions,
[all …]
Dcuda_dnn.cc4061 const dnn::BatchDescriptor& input_dimensions, in DoMatMul() argument
4064 if (input_dimensions.count() != output_dimensions.count()) { in DoMatMul()
4077 if (input_dimensions.layout() != dnn::DataLayout::kBatchYXDepth && in DoMatMul()
4078 input_dimensions.layout() != dnn::DataLayout::kBatchDepthYX) { in DoMatMul()
4103 const int64 n = input_dimensions.count(); in DoMatMul()
4104 const int64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
4148 const uint64 n = input_dimensions.count(); in DoMatMul()
4149 const uint64 k = input_dimensions.NodesAcrossFeatureMaps(); in DoMatMul()
4159 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul()
4263 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc748 const dnn::BatchDescriptor &input_dimensions, in ThenMatMul() argument
751 VLOG_CALL(PARAM(input_data), PARAM(weights), PARAM(input_dimensions), in ThenMatMul()
755 CheckError(dnn->DoMatMul(this, input_data, weights, input_dimensions, in ThenMatMul()
766 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized() argument
770 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
775 input_dimensions, output_dimensions, in ThenMatMulQuantized()
786 const dnn::BatchDescriptor &input_dimensions, in ThenMatMulQuantized() argument
790 PARAM(input_dimensions), PARAM(output_dimensions), in ThenMatMulQuantized()
795 input_dimensions, output_dimensions, in ThenMatMulQuantized()
821 const dnn::BatchDescriptor &input_dimensions, in ThenPoolForward() argument
[all …]
Ddnn.h1566 const dnn::BatchDescriptor& input_dimensions,
1578 const dnn::BatchDescriptor& input_dimensions,
1590 const dnn::BatchDescriptor& input_dimensions,
1638 const dnn::BatchDescriptor& input_dimensions,
1646 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
1657 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
1668 const dnn::BatchDescriptor& input_dimensions, in DoPoolForward() argument
1680 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward() argument
1693 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward() argument
1706 const dnn::BatchDescriptor& input_dimensions, in DoPoolBackward() argument
[all …]
Dstream.h517 const dnn::BatchDescriptor &input_dimensions,
524 const dnn::BatchDescriptor &input_dimensions,
531 const dnn::BatchDescriptor &input_dimensions,
541 const dnn::BatchDescriptor &input_dimensions,
548 const dnn::BatchDescriptor &input_dimensions,
555 const dnn::BatchDescriptor &input_dimensions,
562 const dnn::BatchDescriptor &input_dimensions,
569 const dnn::BatchDescriptor &input_dimensions,
578 const dnn::BatchDescriptor &input_dimensions,
587 const dnn::BatchDescriptor &input_dimensions,
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dquantize_and_dequantize_op.cc167 absl::Span<const int64> input_dimensions = input_shape.dimensions(); in Compile() local
169 return xla::BroadcastInDim(op, input_dimensions, {axis_}); in Compile()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dmetal_delegate.mm363 std::map<::tflite::gpu::ValueId, BHWC> input_dimensions;
372 input_dimensions[input] = input_tensor.shape;
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DREADME.md1379 ceil(input_dimensions[i] / strides[i]).
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2813 `∀ i . 0 ≤ i < rank ⇒ input_dimensions[permutation[i]] = output_dimensions[i]`.