Searched refs:input_squared (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 455 auto input_squared = local
469 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims,
555 auto input_squared = local
569 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims,
666 auto input_squared = local
680 ReferenceUtil::Reduce4DTo1D(*input_squared, /*init=*/0.0f, reduce_dims,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | llvm_ir_runtime.cc | 78 llvm::Value* input_squared = vsl.Mul(input_clamped, input_clamped); local
82 vsl.MulAdd(input_squared, numerator, GetIeeeF32(numerator_coeffs[i]));
89 denominator = vsl.MulAdd(input_squared, denominator,
|
Completed in 181 milliseconds