Home
last modified time | relevance | path

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

123

/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/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dobject_manager.cc29 std::vector<float> transposed(GetElementsSizeForPHWC4(tensor.shape)); in CreatePHWC4BufferFromTensor() local
31 ConvertToPHWC4(tensor.data, tensor.shape, absl::MakeSpan(transposed))); in CreatePHWC4BufferFromTensor()
32 return CreateReadOnlyShaderStorageBuffer<float>(transposed, gl_buffer); in CreatePHWC4BufferFromTensor()
/external/llvm-project/llvm/test/Transforms/LowerMatrixIntrinsics/
Dpropagate-mixed-users.ll45 %transposed = call <8 x double> @llvm.matrix.transpose(<8 x double> %in, i32 4, i32 2)
46 store <8 x double> %transposed, <8 x double>* %Ptr, align 8
47 call void @foo(<8 x double> %transposed)
48 ret <8 x double> %transposed
/external/tensorflow/tensorflow/lite/g3doc/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/python/keras/layers/
Dmerge.py140 transposed = False
156 transposed = True
160 transposed = True
166 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/llvm-project/mlir/integration_test/Dialect/Vector/CPU/
Dtest-transpose.mlir62 // transposed matrices:
110 // transposed tensors:
/external/eigen/bench/
Dsparse_transpose.cpp84 BENCH(for (int k=0; k<REPEAT; ++k) gmm::copy(gmm::transposed(m1),m3);) in main()
/external/libvpx/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.S39 @ transposed rows) and then transpose the results (so that it goes back
79 @ do the transform on transposed rows
/external/mesa3d/src/compiler/spirv/
Dvtn_alu.c69 struct vtn_ssa_value *src0_transpose = wrap_matrix(b, _src0->transposed); in matrix_multiply()
70 struct vtn_ssa_value *src1_transpose = wrap_matrix(b, _src1->transposed); in matrix_multiply()
191 if (src0->transposed) { in vtn_handle_matrix_alu()
192 return vtn_ssa_transpose(b, mat_times_scalar(b, src0->transposed, in vtn_handle_matrix_alu()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_patterns.td46 // transposed b.
55 // transposed.
/external/eigen/unsupported/Eigen/src/Skyline/
DSkylineInplaceLU.h100 const int transposed = 0) const;
307 …xType>::solve(const MatrixBase<BDerived> &b, MatrixBase<XDerived>* x, const int transposed) const { in solve() argument
/external/llvm-project/llvm/test/CodeGen/Hexagon/autohvx/
Disel-shuff-single.ll4 ; transposed into the other vector before the generated shuffles can take

123