Cross Reference: /external/tensorflow/tensorflow/compiler/xla/service/batchnorm

Lines Matching defs:ptype
164   PrimitiveType ptype = operand_shape.element_type();
171                       elements_per_feature_literal->Convert(ptype));
180   TF_ASSIGN_OR_RETURN(zero_literal, zero_literal->Convert(ptype));
184   TF_ASSIGN_OR_RETURN(epsilon_literal, epsilon_literal->Convert(ptype));
202       GetScalarBinaryComputation(ptype, HloOpcode::kAdd);
258   TF_ASSIGN_OR_RETURN(neg_half_literal, neg_half_literal->Convert(ptype));
313   PrimitiveType ptype = operand_shape.element_type();
322   TF_ASSIGN_OR_RETURN(epsilon_literal, epsilon_literal->Convert(ptype));
359   TF_ASSIGN_OR_RETURN(neg_half_literal, neg_half_literal->Convert(ptype));
429   PrimitiveType ptype = activation_shape.element_type();
443                       elements_per_feature_literal->Convert(ptype));
448   TF_ASSIGN_OR_RETURN(zero_literal, zero_literal->Convert(ptype));
452   TF_ASSIGN_OR_RETURN(neg_half_literal, neg_half_literal->Convert(ptype));
457   TF_ASSIGN_OR_RETURN(epsilon_literal, epsilon_literal->Convert(ptype));
501       GetScalarBinaryComputation(ptype, HloOpcode::kAdd);
OpenGrok
