Home
last modified time | relevance | path

Searched refs:transposed (Results 1 – 25 of 76) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dconvert.cc98 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4() local
100 absl::MakeSpan(transposed.data(), transposed.size()), in ConvertToPHWO4I4()
103 return transposed; in ConvertToPHWO4I4()
108 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4Transposed() local
110 absl::MakeSpan(transposed.data(), transposed.size()), in ConvertToPHWO4I4Transposed()
113 return transposed; in ConvertToPHWO4I4Transposed()
164 std::vector<float> transposed(GetElementsSizeForPHWO4I4(tensor.shape)); in ConvertToPHWO4I4() local
166 absl::MakeSpan(transposed.data(), transposed.size())) in ConvertToPHWO4I4()
168 return transposed; in ConvertToPHWO4I4()
210 std::vector<float> transposed(GetElementsSizeForPIOHW4(tensor.shape)); in ConvertToPIOHW4() local
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseMatrixSparseMatMul.pbtxt18 description: "Indicates whether `a` should be transposed."
22 description: "Indicates whether `b` should be transposed."
26 description: "Indicates whether `a` should be conjugate-transposed."
30 description: "Indicates whether `b` should be conjugate-transposed."
35 `b`; returns a sparse matrix `a * b`, unless either `a` or `b` is transposed or
38 Each matrix may be transposed or adjointed (conjugated and transposed)
45 according to whether either `a` or `b` is transposed or adjointed.
106 transpose_a: If True, `a` transposed before multiplication.
107 transpose_b: If True, `b` transposed before multiplication.
Dapi_def_MatMul.pbtxt6 If true, "a" is transposed before multiplication.
12 If true, "b" is transposed before multiplication.
18 "a" (after being transposed if transpose_a is true) must match the
19 outer dimension of "b" (after being transposed if transposed_b is
Dapi_def_QuantizedMatMulWithBias.pbtxt20 transposed if `transposed_b` is non-zero).
61 description: "If true, `a` is transposed before multiplication."
65 description: "If true, `b` is transposed before multiplication."
79 dimension of `a` (after being transposed if `transpose_a` is non-zero) must
80 match the outer dimension of `b` (after being transposed if `transposed_b` is
Dapi_def_QuantizedMatMulWithBiasAndRelu.pbtxt20 transposed if `transposed_b` is non-zero).
61 description: "If true, `a` is transposed before multiplication."
65 description: "If true, `b` is transposed before multiplication."
79 dimension of `a` (after being transposed if `transpose_a` is non-zero) must
80 match the outer dimension of `b` (after being transposed if `transposed_b` is
Dapi_def_SparseMatrixMatMul.pbtxt18 description: "Indicates whether `a` should be transposed."
22 description: "Indicates whether `b` should be transposed."
26 description: "Indicates whether `a` should be conjugate-transposed."
30 description: "Indicates whether `b` should be conjugate-transposed."
Dapi_def_QuantizedMatMul.pbtxt54 If true, `a` is transposed before multiplication.
60 If true, `b` is transposed before multiplication.
73 `a` (after being transposed if `transpose_a` is non-zero) must match the
74 outer dimension of `b` (after being transposed if `transposed_b` is
Dapi_def_QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt20 transposed if `transposed_b` is non-zero).
74 description: "If true, `a` is transposed before multiplication."
78 description: "If true, `b` is transposed before multiplication."
92 dimension of `a` (after being transposed if `transpose_a` is non-zero) must
93 match the outer dimension of `b` (after being transposed if `transposed_b` is
/external/eigen/bench/btl/libs/gmm/
Dgmm_interface.hh86 gmm::mult(gmm::transposed(A),gmm::transposed(B), X); in transposed_matrix_matrix_product()
90 gmm::mult(gmm::transposed(A),A, X); in ata_product()
94 gmm::mult(A,gmm::transposed(A), X); in aat_product()
102 gmm::mult(gmm::transposed(A),B,X); in atv_product()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtranspose_op.cc80 xla::XlaOp transposed; in Compile() local
83 transposed = ctx->Input("x"); in Compile()
85 transposed = xla::Transpose(ctx->Input("x"), transposed_order); in Compile()
90 ctx->SetOutput(0, xla::Conj(transposed)); in Compile()
92 ctx->SetOutput(0, transposed); in Compile()
/external/ComputeLibrary/tests/validation/reference/
DGEMMInterleaveBlocked.h50 …ve_blocked(const SimpleTensor<T> &in, SimpleTensor<T> &out, int int_by, int block, bool transposed) in gemm_interleave_blocked() argument
63 if(!transposed) in gemm_interleave_blocked()
77 …nst SimpleTensor<uint8_t> &in, SimpleTensor<uint8_t> &out, int int_by, int block, bool transposed);
DDFT.cpp348 auto transposed = permute(first_pass, PermutationVector(1U, 0U)); in rdft_2d() local
349 auto second_pass = dft_1d_core(transposed, direction); in rdft_2d()
359 auto transposed = permute(src, PermutationVector(1U, 0U)); in ridft_2d() local
360 auto first_pass = dft_1d_core(transposed, direction); in ridft_2d()
377 auto transposed = permute(first_pass, PermutationVector(1U, 0U)); in dft_2d() local
378 auto second_pass = dft_1d_core(transposed, direction); in dft_2d()
383 auto transposed = permute(src, PermutationVector(1U, 0U)); in dft_2d() local
384 auto first_pass = dft_1d_core(transposed, direction); in dft_2d()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dobject_manager.cc32 std::vector<float> transposed(GetElementsSizeForPHWC4(tensor.shape)); in CreatePHWC4BufferFromTensor() local
34 ConvertToPHWC4(tensor.data, tensor.shape, absl::MakeSpan(transposed))); in CreatePHWC4BufferFromTensor()
35 return CreateReadOnlyShaderStorageBuffer<float>(transposed, gl_buffer); in CreatePHWC4BufferFromTensor()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2022-093.md1 ## TFSA-2022-093: Segfault TFLite converter on per-channel quantized transposed convolutions
7 When converting transposed convolutions using per-channel weight quantization the converter segfaul…
/external/armnn/src/armnn/test/optimizations/
DMoveTransposeUpTests.cpp19 const armnn::TensorInfo transposed({ 1, 3, 5, 2 }, armnn::DataType::Float32); variable
34 head->GetOutputHandler().SetTensorInfo(transposed);
/external/tensorflow/tensorflow/dtensor/python/
Dtpu_util.py587 transposed = []
589 transposed += [
595 transposed[:num_rings], rotate=rotate_ring_across_rings)
597 transposed[r:r + num_rings] = [
598 transposed[r + permutation[i]] for i in range(num_rings)
603 untransposed += [transposed[g + i] for g in range(0, num_cores, num_rings)]
/external/tensorflow/tensorflow/lite/g3doc/examples/convert/
Drnn.md84 1. The **weight** and **recurrent\_kernel** tensors are transposed.
85 1. The transposed weight, transposed recurrent\_kernel and **bias** tensors are
170 1. **Weight** and **projection** tensors are transposed.
173 gate}** are extracted by slicing the transposed weight tensor.
178 1. The **projection** is extracted by slicing the transposed projection tensor.
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtranspose_test.cc120 Array2D<float> transposed({{1.0f, 3.0f, 5.0f}, {2.0f, 4.0f, 6.0f}}); in TEST_F() local
128 const Array2D<float>& expected = transposes % 2 == 0 ? input : transposed; in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc818 Shape transposed = ShapeUtil::MakeShapeWithLayout(F32, {64, 128}, {0, 1}); in TEST() local
819 EXPECT_EQ(std::nullopt, ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
824 Shape transposed = ShapeUtil::MakeShapeWithLayout(F32, {128, 64}, {0, 1}); in TEST() local
826 ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
831 Shape transposed = in TEST() local
834 ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
840 Shape transposed = in TEST() local
843 ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
/external/tensorflow/tensorflow/python/keras/layers/
Dmerge.py136 transposed = False
152 transposed = True
156 transposed = True
162 if transposed:
/external/mesa3d/src/gallium/auxiliary/vl/
Dvl_idct.c74 bool transposed, float size) in calc_addr() argument
76 unsigned wm_start = (right_side == transposed) ? TGSI_WRITEMASK_X : TGSI_WRITEMASK_Y; in calc_addr()
79 unsigned wm_tc = (right_side == transposed) ? TGSI_WRITEMASK_Y : TGSI_WRITEMASK_X; in calc_addr()
97 struct ureg_src saddr[2], bool right_side, bool transposed, in increment_addr() argument
100 unsigned wm_start = (right_side == transposed) ? TGSI_WRITEMASK_X : TGSI_WRITEMASK_Y; in increment_addr()
101 unsigned wm_tc = (right_side == transposed) ? TGSI_WRITEMASK_Y : TGSI_WRITEMASK_X; in increment_addr()
/external/eigen/doc/
DCustomizingEigen_Plugins.dox42 inline Transpose<Derived> transposed() {return this->transpose();}
43 inline const Transpose<Derived> transposed() const {return this->transpose();}
/external/eigen/bench/
Dsparse_transpose.cpp84 BENCH(for (int k=0; k<REPEAT; ++k) gmm::copy(gmm::transposed(m1),m3);) in main()
/external/libvpx/vpx_dsp/arm/
Didct4x4_add_neon.asm32 ; transposed rows) and then transpose the results (so that it goes back
72 ; do the transform on transposed rows
/external/libvpx/config/arm-neon/vpx_dsp/arm/
Didct4x4_add_neon.asm.S38 @ transposed rows) and then transpose the results (so that it goes back
78 @ do the transform on transposed rows

1234