| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
| D | multiple_additive_trees_test.cc | 59 auto output_matrix = output_tensor.matrix<float>(); in TEST_F() local
65 &threads, output_matrix, in TEST_F()
67 EXPECT_EQ(0, output_matrix(0, 0)); in TEST_F()
68 EXPECT_EQ(0, output_matrix(1, 0)); in TEST_F()
95 auto output_matrix = output_tensor.matrix<float>(); in TEST_F() local
103 batch_features_, &threads, output_matrix, in TEST_F()
105 EXPECT_FLOAT_EQ(-0.2f, output_matrix(0, 0)); // -0.4 (bias) + 0.2 (leaf 2). in TEST_F()
106 EXPECT_FLOAT_EQ(0.5f, output_matrix(1, 0)); // -0.4 (bias) + 0.9 (leaf 1). in TEST_F()
115 batch_features_, &threads, output_matrix, in TEST_F()
117 EXPECT_FLOAT_EQ(-0.2f, output_matrix(0, 0)); // -0.4 (bias) + 0.2 (leaf 2). in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 55 MatrixDescriptor output_matrix, double alpha, double beta, in DoGemm() argument
57 DCHECK(!output_matrix.transpose); in DoGemm()
61 CHECK_EQ(batch_size, output_matrix.batch_size); in DoGemm()
64 se::DeviceMemory<Element> output_data(output_matrix.data); in DoGemm()
75 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemm()
76 output_matrix.num_cols, /*size of reduce dim=*/k, /*alpha=*/alpha, in DoGemm()
79 &output_data, /*leading dim of output=*/output_matrix.num_rows) in DoGemm()
85 int64 output_stride = output_matrix.num_rows * output_matrix.num_cols; in DoGemm()
88 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemm()
89 output_matrix.num_cols, /*size of reduce dim=*/k, in DoGemm()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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()
|
| 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 | 196 auto output_matrix = output_matrix_t->matrix<float>(); in Compute() local
259 output_matrix(new_row, new_col) = loaded_tensor(row, old_col); in Compute()
284 output_matrix(i, j) = initializing_values(initializing_values_index); in Compute()
|
| D | scatter_nd_op.cc | 690 auto output_matrix = in DoScatterNd() local
706 output_matrix, indices_flat, updates_flat, output_matrix); \ in DoScatterNd()
|
| /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/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1195 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
1202 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1295 gemmlowp::MatrixMap<int16, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
1316 gemm_context, weights_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
2440 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in Conv()
2447 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in Conv()
2626 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in ConvAsGemm()
2633 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in ConvAsGemm()
3896 gemmlowp::MatrixMap<int16, gemmlowp::MapOrder::ColMajor> output_matrix( in LstmCell()
3912 gemm_context, weights_matrix, input_matrix, &output_matrix, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1214 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
1221 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1287 gemmlowp::MatrixMap<int16, gemmlowp::MapOrder::ColMajor> output_matrix( in FullyConnected()
1308 gemm_context, weights_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
2458 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in Conv()
2465 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in Conv()
3780 gemmlowp::MatrixMap<int16, gemmlowp::MapOrder::ColMajor> output_matrix( in LstmCell()
3796 gemm_context, weights_matrix, input_matrix, &output_matrix, in LstmCell()
|
| D | legacy_optimized_ops.h | 771 gemmlowp::MatrixMap<uint8, gemmlowp::MapOrder::ColMajor> output_matrix( in ConvAsGemm()
778 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in ConvAsGemm()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 36216 …ut, num_rows int64, num_cols int64, optional ...LoadAndRemapMatrixAttr) (output_matrix tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 16539 name: "output_matrix"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 32539 name: "output_matrix"
|
| D | ops_history.v1.pbtxt | 32924 name: "output_matrix"
|