Home
last modified time | relevance | path

Searched defs:should_cast (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dresult_caster_test.cc54 const bool should_cast = result_type != accumulation_type; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_partition.h67 ShouldCastFunc should_cast = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/
H A Dd_variable.py105 def should_cast(self, v): member in _DVariableSaveable
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_mixed_precision.cc2092 bool should_cast = false; in InsertCastNodeAtFanout() local