Searched defs:xla_type (Results 1 – 5 of 5) sorted by relevance
90 xla::PrimitiveType xla_type; in GetOrCreateMax() local106 xla::PrimitiveType xla_type; in GetOrCreateMin() local122 xla::PrimitiveType xla_type; in GetOrCreateAdd() local138 xla::PrimitiveType xla_type; in GetOrCreateMul() local
68 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); in Compile() local117 XlaOpKernelContext* ctx, DataType type, xla::PrimitiveType xla_type, in CrossEntropyWithLogits()178 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); in Compile() local
61 xla::PrimitiveType xla_type; in Compile() local
54 xla::PrimitiveType xla_type) { in BilinearWeights()169 xla::PrimitiveType xla_type) { in ScatterToGradData()
1117 #define ELEMENTS_ATTR_TO_LITERAL(xla_type, cpp_type) \ in CreateArrayLiteralFromAttr() argument