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.cc135 struct SumOp { struct
137 …n(OpKernelContext *ctx, typename TTypes<T>::Scalar &s, const typename TTypes<T>::UnalignedVec &v) { in Run()
140 static StringPiece Name() { in Name()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduction_ops.cc30 class SumOp : public XlaReductionOp { class
32 explicit SumOp(OpKernelConstruction* ctx) in SumOp() function in tensorflow::__anon4d8315190111::SumOp
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dreduce.h262 struct SumOp { struct
269 struct CastSumOp { argument