Home
last modified time | relevance | path

Searched refs:MaybeConvertF32ToBF16 (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc36 xla::XlaOp MaybeConvertF32ToBF16(xla::XlaOp input, DataType dtype) { in MaybeConvertF32ToBF16() function
92 uniform = MaybeConvertF32ToBF16(uniform, dtype_); in Compile()
181 normal = MaybeConvertF32ToBF16(normal, dtype_); in Compile()
223 truncated_normal = MaybeConvertF32ToBF16(truncated_normal, dtype_); in Compile()
Drandom_ops_util.h30 xla::XlaOp MaybeConvertF32ToBF16(xla::XlaOp input, DataType dtype);
Dstateful_random_ops.cc272 normal = MaybeConvertF32ToBF16(normal, dtype_); in Compile()