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

  /external/tensorflow/tensorflow/core/kernels/
segment_reduction_ops.cc 117 Eigen::DSizes<Eigen::DenseIndex, 1> dims_to_reduce; variable
118 dims_to_reduce[0] = 0;
120 Eigen::IndexList<Eigen::type2index<0> > dims_to_reduce; variable
188 out_slice = in_slice.reduce(dims_to_reduce, Reducer());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc     [all...]

Completed in 1389 milliseconds