| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 99 transpose_output = op.get_attr("transpose_output")
109 if not transpose_output:
118 grad_b = sparse_matmul(a, grad, transpose_a=not t_a, transpose_output=t_b)
128 transpose_output=adj_b,
132 grad_b = sparse_matmul(a, grad, transpose_output=True)
136 conj(a), grad, transpose_output=True, conjugate_output=True)
147 a, grad, transpose_a=not t_a, transpose_b=True, transpose_output=t_b)
158 transpose_output=adj_b,
162 grad_b = sparse_matmul(a, grad, transpose_b=True, transpose_output=True)
165 grad_b = sparse_matmul(a, grad, adjoint_b=True, transpose_output=True)
|
| D | sparse_csr_matrix_ops.py | 231 transpose_output=True)
238 transpose_output=True,
244 transpose_output=True,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseMatrixMatMul.pbtxt | 33 name: "transpose_output"
45 If transpose_output is false, returns:
50 If transpose_output is `true`, returns:
62 If both conjugate_output and transpose_output are `true`, returns:
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 63 transpose_output, conjugate_output): argument
92 transpose_output=transpose_output,
|
| D | csr_sparse_matrix_ops_test.py | 586 transpose_output=False,
648 transpose_output=True,
1390 transpose_output=True)
1461 transpose_output=False)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 797 bool transpose_output, MatrixMap* output);
807 bool transpose_output, MatrixMap* output);
945 bool transpose_output, MatrixMap* output);
1025 bool transpose_output = false; in Compute() local
1035 transpose_output = !transpose_output; in Compute()
1047 if (transpose_output) { in Compute()
1057 if (transpose_output) { in Compute()
1061 transpose_output, &out); in Compute()
1066 transpose_output, &out); in Compute()
1089 bool transpose_output, MatrixMap* output) { in ComputeOutputBlock() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | quantize.h | 84 bool transpose_output = false) {
171 if (transpose_output) {
|
| D | quantize_test.cc | 81 bool transpose_output = false) { in GenerateLargeSizeMinCombinedOutput() argument
90 if (transpose_output) { in GenerateLargeSizeMinCombinedOutput()
117 if (transpose_output) { in GenerateLargeSizeMinCombinedOutput()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseMatrixMatMul.pbtxt | 48 name: "transpose_output"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops_test.cc | 109 bool adjoint_b, bool transpose_output) { in TEST() argument
117 .Attr("transpose_output", transpose_output) in TEST()
|
| D | sparse_csr_matrix_ops.cc | 247 bool transpose_output = false; in __anond96a6a670702() local
252 TF_RETURN_IF_ERROR(c->GetAttr("transpose_output", &transpose_output)); in __anond96a6a670702()
271 if (transpose_output) { in __anond96a6a670702()
|
| D | ops.pbtxt | 43750 name: "transpose_output"
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 238 const int64 num_cols, const bool transpose_output, in AllocateOutput() argument
244 if (!transpose_output) { in AllocateOutput()
722 explicit CSRSparseMatrixMatMul(const bool transpose_output) in CSRSparseMatrixMatMul() argument
723 : transpose_output_(transpose_output) {} in CSRSparseMatrixMatMul()
|
| D | kernels.h | 147 explicit CSRSparseMatrixMatMul(const bool transpose_output);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 625 const string transpose_output = in ConvertFullyConnectedOperator() local
628 AvailableArrayName(model, transpose_output + "/perm"); in ConvertFullyConnectedOperator()
632 transpose_op->set_name(transpose_output); in ConvertFullyConnectedOperator()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4013 …, \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'transpose_output\', \'conjugat…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4013 …, \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'transpose_output\', \'conjugat…
|