Searched defs:warp_type (Results 1 – 1 of 1) sorted by relevance
196 xla::PrimitiveType warp_type, TensorShape warp_shape, in BoundSamples()250 xla::PrimitiveType warp_type, TensorShape warp_shape, in CalculateGradData()509 xla::PrimitiveType warp_type = ctx->input_xla_type(1); in Compile() local630 xla::PrimitiveType warp_type = ctx->input_xla_type(1); in Compile() local