Searched refs:sum_tensor (Results 1 – 2 of 2) sorted by relevance
80 Tensor sum_tensor(DT_FLOAT, TensorShape({channels})); in ResizeAreaBaseline() local81 typename TTypes<float, 1>::Tensor sum_data = sum_tensor.vec<float>(); in ResizeAreaBaseline()
255 Tensor* sum_tensor = nullptr; in AddToAccumGradFunction() local263 &sum_tensor)); in AddToAccumGradFunction()264 auto sum_flat = sum_tensor->flat_outer_dims<T>(); in AddToAccumGradFunction()311 accum_val_ = sum_tensor; in AddToAccumGradFunction()