Searched refs:FCFCAdd (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 36 FCFCAdd::FCFCAdd(const OperationDef& definition, const GpuInfo& gpu_info) in FCFCAdd() function in tflite::gpu::FCFCAdd
58 FCFCAdd::FCFCAdd(FCFCAdd&& kernel) : GPUOperation(std::move(kernel)) {} in FCFCAdd() function in tflite::gpu::FCFCAdd
60 FCFCAdd& FCFCAdd::operator=(FCFCAdd&& kernel) { in operator =()
73 std::string FCFCAdd::GetFCFCAddKernelCode(const OperationDef& op_def, in GetFCFCAddKernelCode()
171 int3 FCFCAdd::GetGridSize() const { return int3(dst_[0]->Slices(), 1, 1); } in GetGridSize()
173 FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info, const OperationDef& definition, in CreateFCFCAdd()
176 FCFCAdd result(definition, gpu_info); in CreateFCFCAdd()
|
| D | fc_fc_add.h | 92 class FCFCAdd : public GPUOperation {
94 FCFCAdd() = default;
104 FCFCAdd(FCFCAdd&& kernel);
105 FCFCAdd& operator=(FCFCAdd&& kernel);
106 FCFCAdd(const FCFCAdd&) = delete;
107 FCFCAdd& operator=(const FCFCAdd&) = delete;
110 FCFCAdd(const OperationDef& definition, const GpuInfo& gpu_info);
111 friend FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info,
125 void FCFCAdd::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights()
174 FCFCAdd CreateFCFCAdd(const GpuInfo& gpu_info, const OperationDef& definition,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | special_selector.cc | 176 FCFCAdd fc = CreateFCFCAdd(gpu_info, op_def, fc0_attr, fc1_attr); in TryFCFCAdd()
177 *gpu_op = absl::make_unique<FCFCAdd>(std::move(fc)); in TryFCFCAdd()
|