Home
last modified time | relevance | path

Searched refs:mutable_data (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dcuda_solvers.cc375 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in PotrfImpl()
408 CUDAComplex(dev_workspace.mutable_data()), dev_pivots, dev_lapack_info)); in GetrfImpl()
468 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in GeqrfImpl()
505 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); in UnmqrImpl()
548 CUDAComplex(dev_tau), CUDAComplex(dev_workspace.mutable_data()), in UngqrImpl()
589 CUDAComplex(dev_W), CUDAComplex(dev_workspace.mutable_data()), in HeevdImpl()
627 CUDAComplex(dev_workspace.mutable_data()), in GesvdImpl()
667 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
673 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in GetrfBatchedImpl()
674 dev_pivots, dev_lapack_info->mutable_data(), batch_size)); in GetrfBatchedImpl()
[all …]
Dqr_op_impl.h210 dev_info.back().mutable_data() + batch), in ComputeAsync()
260 dev_info.back().mutable_data() + batch), in ComputeAsync()
277 &tau_matrix(batch, 0), dev_info.back().mutable_data() + batch), in ComputeAsync()
Dmatrix_solve_op.cc221 reinterpret_cast<const Scalar**>(input_copy_ptrs.mutable_data()); in ComputeAsync()
288 reinterpret_cast<const Scalar**>(input_copy_ptr_array.mutable_data()); in ComputeAsync()
291 transposed_rhs_ptr_array.mutable_data()); in ComputeAsync()
Dmatrix_inverse_op.cc173 reinterpret_cast<const Scalar**>(input_copy_ptr_array.mutable_data()); in ComputeAsync()
175 reinterpret_cast<const Scalar**>(output_ptr_array.mutable_data()); in ComputeAsync()
Ddeterminant_op.cc208 reinterpret_cast<const Scalar**>(input_copy_ptrs.mutable_data()); in ComputeAsync()
240 pivots_mat.data(), output_reshaped, dev_info.back().mutable_data()); in ComputeAsync()
352 reinterpret_cast<const Scalar**>(input_copy_ptrs.mutable_data()); in ComputeAsync()
Dcuda_solvers.h350 Scalar* mutable_data() {
404 stream->ThenMemcpy(copy.mutable_data(), wrapped_src, this->bytes())
Dsvd_op_gpu.cu.cc248 outputU_ptr, outputV_ptr, dev_info.back().mutable_data(), in PerformSVD_MgeqN()
322 outputU_ptr, outputV_ptr, dev_info.back().mutable_data(), in PerformSVD_MlessN()
Dself_adjoint_eig_v2_op_gpu.cc138 dev_info.back().mutable_data() + batch), in ComputeAsync()
Dwhere_op.cc296 ->ThenMemcpy(num_true_host.mutable_data(), num_true_ptr, in ComputeAsyncType()
/external/tensorflow/tensorflow/core/lib/gtl/
Darray_slice_test.cc131 EXPECT_EQ(ptr, v.mutable_data()); in MutableTestHelper()
145 *v.mutable_data() = 4; in MutableTestHelper()
197 int* mutable_data() { return storage.data(); } in mutable_data() function
211 int* mutable_data() { return wrong_storage.data(); } in mutable_data() function
608 accept_pointer(s.mutable_data()); in TEST()
630 accept_pointer(s.mutable_data()); in TEST()
Darray_slice_internal.h52 static decltype(std::declval<C>().mutable_data()) Get(C* v) { in Get()
53 return v->mutable_data(); in Get()
Darray_slice.h300 pointer mutable_data() const { return impl_.data(); } in mutable_data() function
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer.h142 SavedSlice* ss = sts.mutable_data(); in Add()
174 Fill(data, num_elements, ss->mutable_data()); in SaveData()
Dtensor_slice_writer.cc191 Fill(data, num_elements, ss->mutable_data()); in SaveData()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc42 *request.mutable_data() = data.handle(); in Transfer()
296 *request.mutable_data() = data.handle(); in Unregister()
364 *request.mutable_data() = data.handle(); in GetShape()
Dglobal_data.cc31 *request.mutable_data() = handle_; in ~GlobalData()
/external/flatbuffers/include/flatbuffers/
Dgrpc.h58 const uint8_t *mutable_data() const { return GRPC_SLICE_START_PTR(slice_); } in mutable_data() function
69 T *GetMutableRoot() { return flatbuffers::GetMutableRoot<T>(mutable_data()); } in GetMutableRoot()
/external/protobuf/src/google/protobuf/compiler/
Dcommand_line_interface_unittest.cc982 std::swap(descriptor_set.mutable_file()->mutable_data()[0], in Run()
983 descriptor_set.mutable_file()->mutable_data()[1]); in Run()
1013 std::swap(descriptor_set.mutable_file()->mutable_data()[0], in Run()
1014 descriptor_set.mutable_file()->mutable_data()[1]); in Run()
Dparser_unittest.cc1719 descriptor_proto->mutable_nested_type()->mutable_data(); in SortMessages()
1731 file_descriptor_proto->mutable_message_type()->mutable_data(); in SortMessages()
/external/tensorflow/tensorflow/compiler/xla/
Dsparse_index_array.h98 tensorflow::gtl::MutableArraySlice<int64> mutable_data() { return &indices_; } in mutable_data() function
/external/libmojo/mojo/public/cpp/bindings/
Dmessage.h73 uint8_t* mutable_data() { return static_cast<uint8_t*>(buffer_->data()); } in mutable_data() function
/external/protobuf/src/google/protobuf/
Drepeated_field.h157 Element* mutable_data();
432 typename TypeHandler::Type** mutable_data();
769 Element** mutable_data();
1177 inline Element* RepeatedField<Element>::mutable_data() {
1542 inline typename TypeHandler::Type** RepeatedPtrFieldBase::mutable_data() {
1966 inline Element** RepeatedPtrField<Element>::mutable_data() {
1967 return RepeatedPtrFieldBase::mutable_data<TypeHandler>();
Drepeated_field_unittest.cc396 *field.mutable_data() = 2; in TEST()
916 string** data = field.mutable_data(); in TEST()
/external/webrtc/talk/media/base/
Dstreamparams.h221 std::vector<StreamParams>* mutable_data() { return &data_; } in mutable_data() function
/external/libmojo/mojo/public/cpp/bindings/tests/
Dvalidation_unittest.cc157 memcpy(message->mutable_data(), &data[0], data.size()); in ReadTestCase()

12