Home
last modified time | relevance | path

Searched refs:FakeQuantWithMinMaxArgsOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfake_quant_ops.cc55 class FakeQuantWithMinMaxArgsOp class
56 : public UnaryElementWiseOp<float, FakeQuantWithMinMaxArgsOp<Device>> {
58 typedef UnaryElementWiseOp<float, FakeQuantWithMinMaxArgsOp<Device>> Base;
59 explicit FakeQuantWithMinMaxArgsOp(OpKernelConstruction* context) in FakeQuantWithMinMaxArgsOp() function in tensorflow::FakeQuantWithMinMaxArgsOp
141 FakeQuantWithMinMaxArgsOp<CPUDevice>);
157 FakeQuantWithMinMaxArgsOp<GPUDevice>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_quantize_ops.cc85 class FakeQuantWithMinMaxArgsOp : public XlaOpKernel { class
87 explicit FakeQuantWithMinMaxArgsOp(OpKernelConstruction* ctx) in FakeQuantWithMinMaxArgsOp() function in tensorflow::__anon41d017970111::FakeQuantWithMinMaxArgsOp
131 REGISTER_XLA_OP(Name("FakeQuantWithMinMaxArgs"), FakeQuantWithMinMaxArgsOp);