Searched refs:InitTensorDataForReduce (Results 1 – 2 of 2) sorted by relevance
159 inline bool InitTensorDataForReduce(const int* dims, const int num_dims, in InitTensorDataForReduce() function188 if (!InitTensorDataForReduce(output_dims, output_num_dims, init_value, in ReduceGeneric()
523 return reference_ops::InitTensorDataForReduce( in ReduceDispatcher()575 return reference_ops::InitTensorDataForReduce(