HomeSort by relevance Sort by last modified time
    Searched refs:sum_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sparse_conditional_accumulator.h 248 // (2) Copy or sum the non-zero elements into sum_indices and sum_tensor
255 Tensor* sum_tensor = nullptr; variable
263 &sum_tensor));
264 auto sum_flat = sum_tensor->flat_outer_dims<T>();
311 accum_val_ = sum_tensor;
resize_area_op_test.cc 77 Tensor sum_tensor(DT_FLOAT, TensorShape({channels}));
78 typename TTypes<float, 1>::Tensor sum_data = sum_tensor.vec<float>();

Completed in 66 milliseconds