Searched defs:tensor_dtype (Results 1 – 3 of 3) sorted by relevance
206 for (const auto& tensor_dtype : collective_fingerprint.tensor_dtypes_) { in get_dtype_strs() local
64 const DataType& tensor_dtype, in PrepareChunkEventProto()
4071 def accumulate_n(inputs, shape=None, tensor_dtype=None, name=None): argument