Searched refs:Softmax1x1 (Results 1 – 6 of 6) sorted by relevance
25 class Softmax1x1 : public GPUOperation {27 Softmax1x1() = default;28 explicit Softmax1x1(const OperationDef& definition);39 Softmax1x1(Softmax1x1&& kernel);40 Softmax1x1& operator=(Softmax1x1&& kernel);41 Softmax1x1(const Softmax1x1&) = delete;42 Softmax1x1& operator=(const Softmax1x1&) = delete;44 friend Softmax1x1 CreateSoftmax1x1();50 Softmax1x1 CreateSoftmax1x1(const OperationDef& definition);
26 Softmax1x1::Softmax1x1(const OperationDef& definition) in Softmax1x1() function in tflite::gpu::Softmax1x132 Softmax1x1::Softmax1x1(Softmax1x1&& kernel) : GPUOperation(std::move(kernel)) {} in Softmax1x1() function in tflite::gpu::Softmax1x134 Softmax1x1& Softmax1x1::operator=(Softmax1x1&& kernel) { in operator =()41 std::string Softmax1x1::GetSoftmaxKernelCode(const OperationDef& op_def) { in GetSoftmaxKernelCode()128 absl::Status Softmax1x1::BindArguments(ArgumentsBinder* args) { in BindArguments()137 int3 Softmax1x1::GetGridSize() const { in GetGridSize()141 Softmax1x1 CreateSoftmax1x1(const OperationDef& definition) { in CreateSoftmax1x1()142 return Softmax1x1(definition); in CreateSoftmax1x1()
114 Softmax1x1 operation = CreateSoftmax1x1(op_def); in Softmax1x1Test()116 src_tensor, absl::make_unique<Softmax1x1>(std::move(operation)), in Softmax1x1Test()153 Softmax1x1 operation = CreateSoftmax1x1(op_def); in Softmax1x1BigNumberTest()155 src_tensor, absl::make_unique<Softmax1x1>(std::move(operation)), in Softmax1x1BigNumberTest()
30 TEST_F(OpenCLOperationTest, Softmax1x1) { in TEST_F() argument
173 Softmax1x1 operation = CreateSoftmax1x1(op_def); in SelectSoftmax()174 *ptr = absl::make_unique<Softmax1x1>(std::move(operation)); in SelectSoftmax()
97 TEST(SoftmaxTest, Softmax1x1) { in TEST() argument