Home
last modified time | relevance | path

Searched refs:operation_ (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc139 : operation_(std::move(operation.operation_)), in ClOperation()
145 operation_ = std::move(operation.operation_); in operator =()
153 return operation_->AddOperation(operation->operation_.get()); in AddOperation()
157 for (int i = 0; i < operation_->src_tensors_names_.size(); ++i) { in UpdateParams()
159 dynamic_cast<const Tensor*>(operation_->src_[i]); in UpdateParams()
163 RETURN_IF_ERROR(cl_args_.SetObjectRef(operation_->src_tensors_names_[i], in UpdateParams()
166 for (int i = 0; i < operation_->dst_tensors_names_.size(); ++i) { in UpdateParams()
168 dynamic_cast<const Tensor*>(operation_->dst_[i]); in UpdateParams()
172 RETURN_IF_ERROR(cl_args_.SetObjectRef(operation_->dst_tensors_names_[i], in UpdateParams()
175 RETURN_IF_ERROR(operation_->BindArguments(&cl_args_)); in UpdateParams()
[all …]
Dcl_operation.h55 operation_ = std::move(gpu_operation); in Init()
58 GPUOperation& GetGpuOperation() { return *operation_; } in GetGpuOperation()
59 const GPUOperation& GetGpuOperation() const { return *operation_; } in GetGpuOperation()
61 const OperationDef& GetDefinition() const { return operation_->definition_; } in GetDefinition()
70 return queue->Dispatch(kernel_, operation_->work_groups_count_, in AddToQueue()
71 operation_->work_group_size_); in AddToQueue()
82 cl_args_.MoveObjectRefsOut(&operation_->args_); in MoveObjectRefsFromCLToGeneric()
85 cl_args_.MoveObjectRefsIn(&operation_->args_); in MoveObjectRefsFromGenericToCL()
87 void SyncScalarValues() { cl_args_.CopyScalarValues(&operation_->args_); } in SyncScalarValues()
90 std::unique_ptr<GPUOperation> operation_;
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc66 operation_ = std::move(operation); in Init()
70 return operation_->definition_; in GetDefinition()
73 bool ComputeTask::IsLinkable() const { return operation_->IsLinkable(); } in IsLinkable()
76 return operation_->AddOperation(task->operation_.get()); in AddTask()
80 operation_->AssembleCode(device->GetInfo()); in Compile()
82 {operation_->dst_tensors_names_[0], operation_->elementwise_code_}}; in Compile()
83 RETURN_IF_ERROR(metal_args_.Init(linkables, device, &operation_->args_, in Compile()
84 &operation_->code_)); in Compile()
86 operation_->args_.ReleaseCPURepresentation(); in Compile()
88 return CompileProgram(device, operation_->definition_.precision, in Compile()
[all …]
Dcompute_task.h75 std::unique_ptr<GPUOperation> operation_; variable
/external/webrtc/third_party/abseil-cpp/absl/base/internal/
Dexception_safety_testing.h889 : factory_(f), operation_(op), contracts_{WrapContract(contracts)...} {}
899 operation_(t_ptr.get());
925 Operation operation_;
991 return {new_factory, operation_, contracts_};
1023 factory_, operation_,
1067 return Test(operation_);
1080 : factory_(f), operation_(o), contracts_(i) {}
1091 Operation operation_;
/external/abseil-cpp/absl/base/internal/
Dexception_safety_testing.h889 : factory_(f), operation_(op), contracts_{WrapContract(contracts)...} {}
899 operation_(t_ptr.get());
925 Operation operation_;
991 return {new_factory, operation_, contracts_};
1023 factory_, operation_,
1067 return Test(operation_);
1080 : factory_(f), operation_(o), contracts_(i) {}
1091 Operation operation_;
/external/libtextclassifier/abseil-cpp/absl/base/internal/
Dexception_safety_testing.h889 : factory_(f), operation_(op), contracts_{WrapContract(contracts)...} {}
899 operation_(t_ptr.get());
925 Operation operation_;
991 return {new_factory, operation_, contracts_};
1023 factory_, operation_,
1067 return Test(operation_);
1080 : factory_(f), operation_(o), contracts_(i) {}
1091 Operation operation_;
/external/angle/third_party/abseil-cpp/absl/base/internal/
Dexception_safety_testing.h897 : factory_(f), operation_(op), contracts_{WrapContract(contracts)...} {}
907 operation_(t_ptr.get());
933 Operation operation_;
999 return {new_factory, operation_, contracts_};
1031 factory_, operation_,
1075 return Test(operation_);
1088 : factory_(f), operation_(o), contracts_(i) {}
1099 Operation operation_;
/external/openscreen/third_party/abseil/src/absl/base/internal/
Dexception_safety_testing.h889 : factory_(f), operation_(op), contracts_{WrapContract(contracts)...} {}
899 operation_(t_ptr.get());
925 Operation operation_;
991 return {new_factory, operation_, contracts_};
1023 factory_, operation_,
1067 return Test(operation_);
1080 : factory_(f), operation_(o), contracts_(i) {}
1091 Operation operation_;
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/base/internal/
Dexception_safety_testing.h889 : factory_(f), operation_(op), contracts_{WrapContract(contracts)...} {}
899 operation_(t_ptr.get());
925 Operation operation_;
991 return {new_factory, operation_, contracts_};
1023 factory_, operation_,
1067 return Test(operation_);
1080 : factory_(f), operation_(o), contracts_(i) {}
1091 Operation operation_;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc126 operation_ = context->op_kernel().name() + "/EigenAllocator"; in Reinitialize()
151 " bytes for ", operation_))); in allocate()
159 LogMemory::RecordRawAllocation(operation_, step_id_, num_bytes, ret, in allocate()
166 LogMemory::RecordRawDeallocation(operation_, step_id_, buffer, allocator_, in deallocate()
170 new AsyncFreeData(allocator_, buffer, operation_, step_id_); in deallocate()
194 : allocator_(a), address_(p), operation_(o), step_id_(s) {} in AsyncFreeData()
197 const string operation_; member
209 LogMemory::RecordRawDeallocation(data->operation_, data->step_id_, in asyncFree()
216 string operation_; in asyncFree() member in tensorflow::EigenGpuStreamDevice