| /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 | 573 transpose_output=False,
635 transpose_output=True,
1377 transpose_output=True)
1448 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_v2/ |
| D | SparseMatrixMatMul.pbtxt | 48 name: "transpose_output"
|
| /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 __anon59a9d6c30702() local
252 TF_RETURN_IF_ERROR(c->GetAttr("transpose_output", &transpose_output)); in __anon59a9d6c30702()
271 if (transpose_output) { in __anon59a9d6c30702()
|
| /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()
775 explicit CSRSparseMatrixMatMul(const bool transpose_output) in CSRSparseMatrixMatMul() argument
776 : transpose_output_(transpose_output) {} in CSRSparseMatrixMatMul()
|
| D | kernels.h | 147 explicit CSRSparseMatrixMatMul(const bool transpose_output);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | ops.mlir | 1176 …n_range = -0.1 : f32, max_range = 0.1 : f32, mode = "MIN_COMBINED", transpose_output = false} : (t…
1184 …n_range = -0.1 : f32, max_range = 0.1 : f32, mode = "MIN_COMBINED", transpose_output = true} : (te…
1192 …n_range = -0.1 : f32, max_range = 0.1 : f32, mode = "MIN_COMBINED", transpose_output = false} : (t…
1200 …rg) {min_range = -0.1 : f32, max_range = 0.1 : f32, mode = "hello", transpose_output = false} : (t…
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.td | 754 the unpacked input type. If transpose_output is true, will return a tensor
755 of shape [dn * unpack_size, dn-1, ..., d1, d0]. transpose_output is faster
757 transpose_output feature.
|
| D | hlo_ops.td | 1353 BoolAttr:$transpose_output,
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 1642 auto transpose_output = subgraph->tensors[op->outputs[0]].get(); in TEST_F() local
1653 transpose_output->quantization->scale[0]); in TEST_F()
1655 transpose_output->quantization->zero_point[0]); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 764 casted, range, ConvertStringRef(op.mode()), op.transpose_output()); in ExportXlaOp()
767 casted, range, ConvertStringRef(op.mode()), op.transpose_output()); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 630 const std::string transpose_output = in ConvertFullyConnectedOperator() local
633 AvailableArrayName(model, transpose_output + "/perm"); in ConvertFullyConnectedOperator()
637 transpose_op->set_name(transpose_output); in ConvertFullyConnectedOperator()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 425 …n_range = -0.1 : f32, max_range = 0.1 : f32, mode = "MIN_COMBINED", transpose_output = false} : (t…
444 …n_range = -0.1 : f32, max_range = 0.1 : f32, mode = "MIN_COMBINED", transpose_output = false, is_1…
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 562 if (op.transpose_output()) { in Verify()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4317 …, \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'transpose_output\', \'conjugat…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4317 …, \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'transpose_output\', \'conjugat…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 25254 …input tf.Output, min_range float32, max_range float32, mode string, transpose_output bool) (output…
25258 …"min_range": min_range, "max_range": max_range, "mode": mode, "transpose_output": transpose_output}
|