Searched defs:input_copy (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmatrix_inverse_op.cc132 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 Dself_adjoint_eig_v2_op_gpu.cc98 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 Ddeterminant_op.cc172 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 Dmatrix_solve_op.cc175 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 Dsvd_op_gpu.cu.cc101 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 Dscatter_nd_op.cc161 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 Dlithium-allocator.cc824 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