Home
last modified time | relevance | path

Searched refs:float_pad_value (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h63 auto float_pad_value = std::numeric_limits<float>::quiet_NaN(); in CreateFirstWithType() local
64 return CreateR0WithType(type, -float_pad_value, b); in CreateFirstWithType()
72 auto float_pad_value = std::numeric_limits<float>::quiet_NaN(); in CreateLastWithType() local
73 return CreateR0WithType(type, float_pad_value, b); in CreateLastWithType()
Dspmd_partitioner.cc3178 float float_pad_value; in HandleSelectAndScatter() local
3182 float_pad_value = -std::numeric_limits<float>::infinity(); in HandleSelectAndScatter()
3184 float_pad_value = std::numeric_limits<float>::infinity(); in HandleSelectAndScatter()
3189 float_pad_value = std::numeric_limits<float>::infinity(); in HandleSelectAndScatter()
3191 float_pad_value = -std::numeric_limits<float>::infinity(); in HandleSelectAndScatter()
3200 static_cast<bfloat16>(float_pad_value)) in HandleSelectAndScatter()
3201 : LiteralUtil::CreateR0<float>(float_pad_value))); in HandleSelectAndScatter()