| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
| multiple_additive_trees_test.cc | 59 auto output_matrix = output_tensor.matrix<float>(); local
65 &threads, output_matrix);
66 EXPECT_EQ(0, output_matrix(0, 0));
67 EXPECT_EQ(0, output_matrix(1, 0));
94 auto output_matrix = output_tensor.matrix<float>(); local
102 batch_features_, &threads, output_matrix);
103 EXPECT_FLOAT_EQ(-0.2f, output_matrix(0, 0)); // -0.4 (bias) + 0.2 (leaf 2).
104 EXPECT_FLOAT_EQ(0.5f, output_matrix(1, 0)); // -0.4 (bias) + 0.9 (leaf 1).
112 output_matrix);
114 EXPECT_FLOAT_EQ(-0.4f * 6 + 0.2 * 3.2, output_matrix(0, 0))
170 auto output_matrix = output_tensor.matrix<float>(); local
256 auto output_matrix = output_tensor.matrix<float>(); local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| decode_wav_op.cc | 77 auto output_matrix = output->matrix<float>(); variable
94 output_matrix(sample, channel) = output_value;
|
| cholesky_grad.cc | 64 MatrixMap output_matrix = outputs->at(0); variable
70 output_matrix = input_matrix_grad.template triangularView<Eigen::Lower>();
93 output_matrix.block(block_end, 0, trailing_size, block_begin);
97 auto C_bar = output_matrix.block(block_end, block_begin, trailing_size,
103 output_matrix.block(block_begin, block_begin, block_size, block_size);
106 auto R_bar = output_matrix.block(block_begin, 0, block_size, block_begin);
118 output_matrix = (0.5 * (output_matrix + output_matrix.transpose())).eval();
|
| load_and_remap_matrix_op.cc | 193 context->allocate_output("output_matrix",
196 auto output_matrix = output_matrix_t->matrix<float>(); variable
259 output_matrix(new_row, new_col) = loaded_tensor(row, old_col);
284 output_matrix(i, j) = initializing_values(initializing_values_index);
|
| scatter_nd_op.cc | 465 auto output_matrix = local
481 output_matrix, indices_flat, updates_flat, output_matrix); \
|
