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

  /external/tensorflow/tensorflow/contrib/lite/kernels/
concatenation.cc 65 int sum_axis = t0->dims->data[axis]; local
76 sum_axis += t->dims->data[axis];
85 output_size->data[d] = (d == axis) ? sum_axis : t0->dims->data[d];
  /external/tensorflow/tensorflow/python/ops/
linalg_ops.py 556 sum_axis = None if axis is None else axis[0]
557 result = math_ops.reduce_sum(result, sum_axis, keepdims=True)
image_ops_impl.py     [all...]
image_ops_test.py     [all...]

Completed in 161 milliseconds