Searched refs:OutputDataType (Results 1 – 3 of 3) sorted by relevance
64 template <ArrayDataType InputsDataType, ArrayDataType OutputDataType>77 CHECK(output_array.data_type == OutputDataType); in EvaluateBinaryOperatorOnConstantInputs()92 auto& output_data = output_array.GetMutableBuffer<OutputDataType>().data; in EvaluateBinaryOperatorOnConstantInputs()136 DataType<OutputDataType> outval; in EvaluateBinaryOperatorOnConstantInputs()173 #define TOCO_HANDLE_CASE(InputsDataType, OutputDataType) \ in EvaluateBinaryOperatorOnConstantInputs() argument175 output_data_type == OutputDataType) { \ in EvaluateBinaryOperatorOnConstantInputs()176 EvaluateBinaryOperatorOnConstantInputs<InputsDataType, OutputDataType>( \ in EvaluateBinaryOperatorOnConstantInputs()
98 template <DataType TfDataType, typename OutputDataType, typename Kernel,99 typename InputDataType = OutputDataType>139 Tensor result_tensor = ConvertDescriptorToTensor<OutputDataType>( in Compute()
72 using OutputDataType = in SerializeAndDeserialize() typedef85 OutputDataType output_data = in SerializeAndDeserialize()86 reinterpret_cast<OutputDataType>(message.mutable_payload()); in SerializeAndDeserialize()