Searched refs:input_squared (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | math_ops.cc | 51 llvm::Value* input_squared = b->CreateFMul(input_clamped, input_clamped); in EmitFastTanh() local
54 numerator = b->CreateFAdd(b->CreateFMul(input_squared, numerator), in EmitFastTanh()
63 b->CreateFAdd(b->CreateFMul(input_squared, denominator), in EmitFastTanh()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_trigonometric_to_approximation.cc | 120 Value input_squared = in emitApproximation() local
126 loc, rewriter.create<MulFOp>(loc, input_squared, numerator), in emitApproximation()
137 loc, rewriter.create<MulFOp>(loc, input_squared, denominator), in emitApproximation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 562 auto input_squared = in XLA_TEST_P() local
576 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
663 auto input_squared = in XLA_TEST_P() local
677 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
774 auto input_squared = in XLA_TEST_P() local
788 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
|