Home
last modified time | relevance | path

Searched defs:SumOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc133 struct SumOp { struct
135 …n(OpKernelContext *ctx, typename TTypes<T>::Scalar &s, const typename TTypes<T>::UnalignedVec &v) { in Run()
138 static StringPiece Name() { in Name()
Dsegment_reduction_ops.cc407 struct SumOp { struct
408 void operator()(const constMatrixChip<T> data, MatrixChip<T> output) { in operator ()()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduction_ops.cc30 class SumOp : public XlaReductionOp { class
32 explicit SumOp(OpKernelConstruction* ctx) in SumOp() function in tensorflow::__anon48fb755d0111::SumOp