Searched refs:CreateScalarComputation (Results 1 – 3 of 3) sorted by relevance
31 XlaComputation CreateScalarComputation(const string& name, PrimitiveType type, in CreateScalarComputation() function51 return CreateScalarComputation( in CreateScalarAddComputation()57 return CreateScalarComputation( in CreateScalarMultiplyComputation()63 return CreateScalarComputation( in CreateScalarGeComputation()69 return CreateScalarComputation( in CreateScalarMaxComputation()75 return CreateScalarComputation( in CreateScalarMinComputation()81 return CreateScalarComputation( in CreateScalarAndComputation()87 return CreateScalarComputation( in CreateScalarOrComputation()
30 XlaComputation CreateScalarComputation(const string& name, PrimitiveType type,
257 CreateScalarComputation("reducer", input_shape.element_type(), in TorchScatterDense()