Searched refs:customOp (Results 1 – 4 of 4) 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()
105 y.device(d) = x.customOp(VecPermute<T>(dst)); in operator()
293 return input.customOp(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