Home
last modified time | relevance | path

Searched defs:transpose_output (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
H A Dcsr_sparse_matrix_dense_mat_mul_grad_test.py57 transpose_output, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_matmul_op.cc1030 bool transpose_output = false; in Compute() local
1098 bool transpose_output, MatrixMap* output) { in ComputeOutputBlock()
1441 bool transpose_output, MatrixMap* output) { in Compute()
1541 bool transpose_output, MatrixMap* output) { in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dmat_mul_op.cc244 const int64_t num_cols, const bool transpose_output, in AllocateOutput()
782 explicit CSRSparseMatrixMatMul(const bool transpose_output) in CSRSparseMatrixMatMul()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_csr_matrix_ops_test.cc109 bool adjoint_b, bool transpose_output) { in TEST()
H A Dsparse_csr_matrix_ops.cc247 bool transpose_output = false; in __anona88575120702() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dquantize_test.cc83 bool transpose_output = false) { in GenerateLargeSizeMinCombinedOutput()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model_test.cc1765 auto transpose_output = subgraph->tensors[op->outputs[0]].get(); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc635 const std::string transpose_output = in ConvertFullyConnectedOperator() local