Searched defs:xla_type (Results 1 – 4 of 4) sorted by relevance
83 xla::PrimitiveType xla_type; in GetOrCreateMax() local99 xla::PrimitiveType xla_type; in GetOrCreateMin() local115 xla::PrimitiveType xla_type; in GetOrCreateAdd() local131 xla::PrimitiveType xla_type; in GetOrCreateMul() local
52 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); in Compile() local92 XlaOpKernelContext* ctx, DataType type, xla::PrimitiveType xla_type, in CrossEntropyWithLogits()161 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); in Compile() local
56 xla::PrimitiveType xla_type; in Compile() local
54 xla::PrimitiveType xla_type) { in BilinearWeights()169 xla::PrimitiveType xla_type) { in ScatterToGradData()