Searched refs:input_squared (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | math_ops.cc | 39 llvm::Value* input_squared = b->CreateFMul(input_clamped, input_clamped); in EmitFastTanh() local
42 numerator = b->CreateFAdd(b->CreateFMul(input_squared, numerator), in EmitFastTanh()
51 b->CreateFAdd(b->CreateFMul(input_squared, denominator), in EmitFastTanh()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 464 auto input_squared = in XLA_TEST_P() local
478 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
565 auto input_squared = in XLA_TEST_P() local
579 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
676 auto input_squared = in XLA_TEST_P() local
690 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims, in XLA_TEST_P()
|