Searched refs:customOp (Results 1 – 2 of 2) sorted by relevance
44 Tensor<float, 2> result = tensor.customOp(InsertZeros()); in test_custom_unary_op()91 Tensor<float, 3> result = tensor1.customOp(tensor2, BatchMatMul()); in test_custom_binary_op()
803 …const TensorCustomUnaryOp<const CustomUnaryFunc, const Derived> customOp(const CustomUnaryFunc& op… in customOp() function808 …const TensorCustomBinaryOp<const CustomBinaryFunc, const Derived, const OtherDerived> customOp(con… in customOp() function