Searched refs:sum_tensor (Results 1 – 2 of 2) sorted by relevance
237 Tensor* sum_tensor = new Tensor(); in AddToAccumGradFunction() local242 OP_REQUIRES_OK(ctx, ctx->allocate_temp(dtype_, sum_shape, sum_tensor)); in AddToAccumGradFunction()243 auto sum_flat = sum_tensor->flat_outer_dims<T>(); in AddToAccumGradFunction()289 accum_val_.reset(sum_tensor); in AddToAccumGradFunction()
80 Tensor sum_tensor(DT_FLOAT, TensorShape({channels})); in ResizeAreaBaseline() local81 typename TTypes<float, 1>::Tensor sum_data = sum_tensor.vec<float>(); in ResizeAreaBaseline()