Searched defs:input_copy (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | matrix_inverse_op.cc | 132 Tensor input_copy; variable 136 input.shape(), &input_copy), 138 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); 141 device.memcpy(input_copy.flat<Scalar>().data(), 146 context, DoConjugateMatrixTranspose(device, input, &input_copy),
|
H A D | self_adjoint_eig_v2_op_gpu.cc | 98 Tensor input_copy; variable 102 {0}, DataTypeToEnum<Scalar>::value, input.shape(), &input_copy), 108 if (!input.SharesBufferWith(input_copy)) { 111 conj(device, input_copy.flat<Scalar>() /*out*/, 114 device.memcpy(input_copy.flat<Scalar>().data(), 124 // Compute eigen decomposition in-place in input_copy. 127 auto input_copy_reshaped = input_copy.flat_inner_dims<Scalar, 3>(); 149 // Transpose eigenvectors now stored in input_copy in column-major form to 152 context, DoMatrixTranspose(device, input_copy, eigenvectors), done);
|
H A D | determinant_op.cc | 172 Tensor input_copy; variable 176 {0}, DataTypeToEnum<Scalar>::value, input.shape(), &input_copy), 178 if (!input.SharesBufferWith(input_copy)) { 179 d.memcpy(input_copy.flat<Scalar>().data(), input.flat<Scalar>().data(), 182 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); 235 // input_copy by the Getrf{Batched} kernel. 238 const_cast<const Tensor*>(&input_copy) 317 Tensor input_copy; variable 321 {0}, DataTypeToEnum<Scalar>::value, input.shape(), &input_copy), 323 if (!input.SharesBufferWith(input_copy)) { [all...] |
H A D | matrix_solve_op.cc | 175 Tensor input_copy; variable 183 input.shape(), &input_copy), 186 DoMatrixTranspose(device, input, &input_copy), done); 191 {0}, DataTypeToEnum<Scalar>::value, input.shape(), &input_copy), 193 if (!input.SharesBufferWith(input_copy)) { 194 device.memcpy(input_copy.flat<Scalar>().data(), 199 auto input_copy_reshaped = input_copy.template flat_inner_dims<Scalar, 3>();
|
H A D | svd_op_gpu.cu.cc | 101 Tensor input_copy; local 106 TensorShape({batch_size, m}), &input_copy), 109 d.memcpy(input_copy.flat<Scalar>().data(), input_ptr, 162 cfg2D, m, full_matrices_ ? m : p, input_copy.flat<Scalar>().data(), 202 Tensor input_copy; local 207 solver->allocate_scoped_tensor(M.dtype(), input_shape, &input_copy), 210 OP_REQUIRES_OK_ASYNC(context, DoMatrixTranspose(device, M, &input_copy), 235 auto input_reshaped = input_copy.template flat_inner_dims<Scalar, 3>(); 271 Tensor input_copy; local 275 {0}, DataTypeToEnum<Scalar>::value, M.shape(), &input_copy), [all...] |
H A D | scatter_nd_op.cc | 161 const Tensor& input_copy = c->input(0); local 162 copy(c->eigen_device<Device>(), params.flat<T>(), input_copy.flat<T>());
|
/external/v8/src/crankshaft/ |
H A D | lithium-allocator.cc | 824 LUnallocated* input_copy = cur_input->CopyUnconstrained( local 828 AddConstraintsGapMove(gap_index, input_copy, cur_input); 834 LUnallocated* input_copy = cur_input->CopyUnconstrained( local 840 if (RequiredRegisterKind(input_copy->virtual_register()) == 847 AddConstraintsGapMove(gap_index, input_copy, cur_input); 860 LUnallocated* input_copy = cur_input->CopyUnconstrained( local 863 AddConstraintsGapMove(gap_index, input_copy, cur_input); 869 instr->pointer_map()->RecordPointer(input_copy, chunk()->zone());
|
Completed in 1578 milliseconds