| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 84 MatrixDescriptor output_matrix, AlphaType alpha, double beta, in DoGemmWithAlgorithm() argument
87 DCHECK(!output_matrix.transpose); in DoGemmWithAlgorithm()
119 se::DeviceMemory<Element> output_data(output_matrix.data); in DoGemmWithAlgorithm()
132 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemmWithAlgorithm()
133 output_matrix.num_cols, in DoGemmWithAlgorithm()
139 /*leading dim of output=*/output_matrix.num_rows, computation_type, in DoGemmWithAlgorithm()
147 int64 output_stride = output_matrix.num_rows * output_matrix.num_cols; in DoGemmWithAlgorithm()
150 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemmWithAlgorithm()
151 output_matrix.num_cols, /*size of reduce dim=*/k, in DoGemmWithAlgorithm()
156 /*leading dim of output=*/output_matrix.num_rows, output_stride, in DoGemmWithAlgorithm()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | cholesky_grad.cc | 64 MatrixMap output_matrix = outputs->at(0); in ComputeMatrix() local
70 output_matrix = input_matrix_grad.template triangularView<Eigen::Lower>(); in ComputeMatrix()
93 output_matrix.block(block_end, 0, trailing_size, block_begin); in ComputeMatrix()
97 auto C_bar = output_matrix.block(block_end, block_begin, trailing_size, in ComputeMatrix()
103 output_matrix.block(block_begin, block_begin, block_size, block_size); in ComputeMatrix()
106 auto R_bar = output_matrix.block(block_begin, 0, block_size, block_begin); in ComputeMatrix()
118 output_matrix = (0.5 * (output_matrix + output_matrix.transpose())).eval(); in ComputeMatrix()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | transpose_op.cc | 112 CSRSparseMatrix output_matrix; in Compute() local
114 transpose(ctx, conjugate_, *input_matrix, &output_matrix)); in Compute()
116 output_t.scalar<Variant>()() = std::move(output_matrix); in Compute()
152 CSRSparseMatrix* output_matrix) { in operator ()() argument
184 output_row_ptr_t, output_col_ind_t, output_values_t, output_matrix)); in operator ()()
194 if (output_matrix->nnz(i) == 0) { in operator ()()
201 output_matrix->row_pointers_vec(i), output_matrix->col_indices_vec(i), in operator ()()
202 output_matrix->values_vec<T>(i), output_dense_shape}; in operator ()()
|
| D | softmax_op.cc | 64 CSRSparseMatrix output_matrix; in Compute() local
73 logits_matrix->col_indices(), output_values_t, &output_matrix)); in Compute()
78 ctx, softmax(ctx, *logits_matrix, output_matrix.values().vec<T>())); in Compute()
82 output_t.scalar<Variant>()() = std::move(output_matrix); in Compute()
|
| D | sparse_mat_mul_op.cc | 231 const SparseMatrix& output_matrix = output_matrices[batch_idx]; in Compute() local
232 const int64 nnz = output_matrix.nonZeros(); in Compute()
233 std::copy(output_matrix.outerIndexPtr(), in Compute()
234 output_matrix.outerIndexPtr() + num_output_rows + 1, in Compute()
236 std::copy(output_matrix.innerIndexPtr(), in Compute()
237 output_matrix.innerIndexPtr() + nnz, in Compute()
239 std::copy(output_matrix.valuePtr(), in Compute()
240 output_matrix.valuePtr() + nnz, in Compute()
|
| D | kernels.h | 212 CSRSparseMatrix* output_matrix);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | decode_wav_op.cc | 77 auto output_matrix = output->matrix<float>(); in Compute() local
94 output_matrix(sample, channel) = output_value; in Compute()
|
| D | load_and_remap_matrix_op.cc | 195 auto output_matrix = output_matrix_t->matrix<float>(); in Compute() local
258 output_matrix(new_row, new_col) = loaded_tensor(row, old_col); in Compute()
283 output_matrix(i, j) = initializing_values(initializing_values_index); in Compute()
|
| D | scatter_nd_op.cc | 935 auto output_matrix = in DoScatterNd() local
951 output_matrix, indices_flat, updates_flat, output_matrix); \ in DoScatterNd()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LoadAndRemapMatrix.pbtxt | 24 name: "output_matrix"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | LoadAndRemapMatrix.pbtxt | 24 name: "output_matrix"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | fractional_avg_pool_op_test.py | 80 output_matrix = self._AvgPoolAlongRows(input_matrix, col_seq, overlapping)
81 return output_matrix.transpose()
|
| D | fractional_max_pool_op_test.py | 80 output_matrix = self._MaxPoolAlongRows(input_matrix, col_seq, overlapping)
81 return output_matrix.transpose()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LoadAndRemapMatrix.pbtxt | 42 name: "output_matrix"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1138 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
1145 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in FullyConnected()
1696 gemmlowp::MatrixMap<int16, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
1717 gemmlowp_context, weights_matrix, input_matrix, &output_matrix, in FullyConnected()
2155 gemmlowp::MatrixMap<int8, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
2163 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in FullyConnected()
2718 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in Conv()
2725 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in Conv()
2896 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in ConvAsGemm()
2903 gemmlowp_context, filter_matrix, input_matrix, &output_matrix, in ConvAsGemm()
[all …]
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 26960 …ut, num_rows int64, num_cols int64, optional ...LoadAndRemapMatrixAttr) (output_matrix tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 21344 name: "output_matrix"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 31876 name: "output_matrix"
|