Searched refs:x_squared (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 105 auto x_squared = Real(x_after_k * Conj(x_after_k)); in House() local
107 Reduce(x_squared, ScalarLike(x_squared, 0.0), in House()
|
| D | elemental_ir_emitter.cc | 1373 auto x_squared = FMul(x, x); in EmitLog1p() local
1379 for_small_x = FMul(FMul(x, x_squared), for_small_x); in EmitLog1p()
1380 for_small_x = FAdd(FMul(negative_half, x_squared), for_small_x); in EmitLog1p()
1434 auto x_squared = FMul(x, x); in EmitExpm1() local
1435 auto x_squared_over_two = FMul(x_squared, half); in EmitExpm1()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 521 x_squared = [math_ops.square(elem) for elem in x]
522 y = embedding_ops.embedding_lookup(x_squared, ids)
|