Home
last modified time | relevance | path

Searched refs:mutable_data (Results 1 – 25 of 31) 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()
669 ldu, CUDAComplex(V), ldv, CUDAComplex(dev_workspace.mutable_data()), in GesvdjBatchedImpl()
711 if (!CopyHostToDevice(context, dev_a_dev_ptrs.mutable_data() /* dest */, in GetrfBatchedImpl()
717 reinterpret_cast<CudaScalar**>(dev_a_dev_ptrs.mutable_data()), lda, in GetrfBatchedImpl()
[all …]
Dqr_op_impl.h213 dev_info.back().mutable_data() + batch), in ComputeAsync()
263 dev_info.back().mutable_data() + batch), in ComputeAsync()
280 &tau_matrix(batch, 0), dev_info.back().mutable_data() + batch), in ComputeAsync()
Dmatrix_solve_op.cc224 reinterpret_cast<const Scalar**>(input_copy_ptrs.mutable_data()); in ComputeAsync()
287 reinterpret_cast<const Scalar**>(input_copy_ptr_array.mutable_data()); in ComputeAsync()
290 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.h359 Scalar* mutable_data() {
413 stream->ThenMemcpy(copy.mutable_data(), wrapped_src, this->bytes())
Dself_adjoint_eig_v2_op_gpu.cc138 dev_info.back().mutable_data() + batch), in ComputeAsync()
Dlu_op_gpu.cu.cc185 packed_triangular_factors_ptrs.mutable_data()); in ComputeAsync()
Dwhere_op.cc297 ->ThenMemcpy(num_true_host.mutable_data(), num_true_ptr, in ComputeAsyncType()
Dsvd_op_gpu.cu.cc136 int* dev_info_ptr = dev_info.back().mutable_data(); in RunSVD()
/external/libchrome/mojo/public/cpp/bindings/
Dmessage.h101 uint8_t* mutable_data() { return const_cast<uint8_t*>(data()); } in COMPONENT_EXPORT()
113 return reinterpret_cast<internal::MessageHeader*>(mutable_data()); in COMPONENT_EXPORT()
122 return reinterpret_cast<internal::MessageHeaderV1*>(mutable_data()); in COMPONENT_EXPORT()
131 return reinterpret_cast<internal::MessageHeaderV2*>(mutable_data()); in COMPONENT_EXPORT()
/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/libchrome/mojo/public/tools/fuzzers/
Dmojo_parse_message_proto_fuzzer.cc34 memcpy(message.mutable_data(), message_str.data(), message_str.size()); in FuzzMessage()
Dmojo_parse_message_fuzzer.cc24 memcpy(message.mutable_data(), data, size); in FuzzMessage()
/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/flatbuffers/include/flatbuffers/
Dgrpc.h57 const uint8_t *mutable_data() const { return GRPC_SLICE_START_PTR(slice_); } in mutable_data() function
68 T *GetMutableRoot() { return flatbuffers::GetMutableRoot<T>(mutable_data()); } in GetMutableRoot()
/external/tensorflow/tensorflow/compiler/xla/
Dsparse_index_array.h99 absl::Span<int64> mutable_data() { return absl::MakeSpan(indices_); } in mutable_data() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc44 *request.mutable_data() = data.handle(); in Transfer()
452 *request.mutable_data() = data.handle(); in GetShape()
Dlocal_client.cc328 TF_ASSIGN_OR_RETURN(*result.mutable_data(), in TransferToLocalServer()
/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/libchrome/mojo/public/cpp/bindings/tests/
Dvalidation_unittest.cc177 memcpy(message->mutable_data(), &data[0], data.size()); in ReadTestCase()

12