/external/tensorflow/tensorflow/compiler/xla/client/ |
D | padding.cc | 26 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()
|
D | padding.h | 44 Status ValidatePaddingValues(absl::Span<const int64> input_dimensions, 60 absl::Span<const int64> input_dimensions,
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 50 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 …]
|
D | rocm_dnn.cc | 3814 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/ |
D | cuda_dnn.h | 411 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 …]
|
D | cuda_dnn.cc | 4061 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/ |
D | stream.cc | 748 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 …]
|
D | dnn.h | 1566 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 …]
|
D | stream.h | 517 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/ |
D | quantize_and_dequantize_op.cc | 167 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/ |
D | metal_delegate.mm | 363 std::map<::tflite::gpu::ValueId, BHWC> input_dimensions; 372 input_dimensions[input] = input_tensor.shape;
|
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | README.md | 1379 ceil(input_dimensions[i] / strides[i]).
|
/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
D | operation_semantics.md | 2813 `∀ i . 0 ≤ i < rank ⇒ input_dimensions[permutation[i]] = output_dimensions[i]`.
|