Searched refs:math_type (Results 1 – 4 of 4) sorted by relevance
284 static int math_type (lua_State *L) { in math_type() function375 {"type", math_type},
95 MathType math_type = 2; field
748 return proto_.math_type() == AlgorithmProto::TENSOR_OP_MATH;
698 cudnnMathType_t math_type = in set_use_tensor_op_math() local701 CHECK_CUDNN_OK(cudnnSetConvolutionMathType(handle_.get(), math_type)); in set_use_tensor_op_math()1092 cudnnMathType_t math_type = in Create() local1096 CHECK_CUDNN_OK(cudnnSetRNNMatrixMathType(rnn_desc.get(), math_type)); in Create()