/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | relu_op.h | 38 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function 76 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function 97 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function 118 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function 144 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function 175 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function 200 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function 221 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function 242 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function 263 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function
|
H A D | softplus_op.cc | 39 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function in tensorflow::SoftplusOp 62 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function in tensorflow::SoftplusGradOp
|
H A D | softsign_op.cc | 39 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function in tensorflow::SoftsignOp 62 void Operate(OpKernelContext* context, const Tensor& g, const Tensor& a, in Operate() function in tensorflow::SoftsignGradOp
|
H A D | fake_quant_ops.cc | 89 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function in tensorflow::FakeQuantWithMinMaxArgsOp 130 void Operate(OpKernelContext* context, const Tensor& gradient, in Operate() function in tensorflow::FakeQuantWithMinMaxArgsGradientOp
|
H A D | relu_op.cc | 217 void Operate(OpKernelContext* context, const Tensor& input, Tensor* output) { in Operate() function in tensorflow::ReluOp
|
H A D | pad_op.cc | 241 void Operate(OpKernelContext* context, in Operate() function in tensorflow::PadOp
|