Searched defs:out_t (Results 1 – 8 of 8) sorted by relevance
33 #define EXTRACT(in_t, out_t) extract_func_##in_t##_to_##out_t argument35 #define EXTRACT_FUNC(in_t, out_t, k) \ argument
35 int64* out_t = static_cast<int64*>(out); in argmax_float_1d_xla_impl() local
39 int64* out_t = static_cast<int64*>(out); in argmax_float_2d_xla_impl() local
83 Tensor *out_t; in Compute() local
265 Tensor* out_t; in Compute() local
459 auto out_t = out->flat<T>(); in ValidateAndInitializeToDense() local470 auto out_t = out->flat<T>(); in ToDense() local
164 Tensor* out_t; member
165 Tensor* out_t; member