| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_Rsqrt.pbtxt | 2 graph_op_name: "Rsqrt"
4 name: "math.Rsqrt"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unary_ops_composition.cc | 241 REGISTER_COMPUTE_FN(Rsqrt); in UnaryOpsCompositionSupport()
280 REGISTER_COMPUTE_FN_HELPER(Rsqrt, functor::rsqrt<T>);
309 REGISTER_COMPUTE_FN(Rsqrt); in UnaryOpsCompositionSupport()
337 REGISTER_COMPUTE_FN_HELPER(Rsqrt, functor::rsqrt<T>);
371 REGISTER_COMPUTE_FN(Rsqrt); in UnaryOpsCompositionSupport()
409 REGISTER_COMPUTE_FN_HELPER(Rsqrt, functor::rsqrt<T>);
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd.cc | 378 auto c_temp = Rsqrt(one + Square(t)); in MakeJacobi()
385 auto rnorm = Rsqrt(Square(c) + Square(s)); in MakeJacobi()
435 XlaOp tmp = Rsqrt(one + Square(u)); in GetOneSidedJacobiRotation()
523 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in OneSidedJacobiUpdate()
533 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in OneSidedJacobiUpdate()
549 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in OneSidedJacobiUpdate()
559 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in OneSidedJacobiUpdate()
775 Rsqrt(Reduce(Square(result.v), ScalarLike(d, 0.0), in SortBySingularValuesAndPostProcessing()
787 Rsqrt(Reduce(Square(result.u), ScalarLike(d, 0.0), in SortBySingularValuesAndPostProcessing()
|
| D | self_adjoint_eig.cc | 96 auto c_temp = Rsqrt(one + Square(t)); in SymmetricShurDecomposition2x2()
103 auto rnorm = Rsqrt(Square(c) + Square(s)); in SymmetricShurDecomposition2x2()
175 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in Update()
182 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in Update()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unary_ops.cc | 83 XLAJIT_MAKE_UNARY(Rsqrt, xla::Rsqrt(x));
|
| D | batch_norm_op.cc | 84 ctx->SetOutput(4, xla::Rsqrt(xla::Add( in Compile()
194 auto scratch1 = xla::Rsqrt(xla::Add(var, epsilon)); in Compile()
|
| D | training_ops.cc | 265 var = var - grad * lr * xla::Rsqrt(accum); in Compile()
316 xla::XlaOp adagrad_lr = lr * xla::Rsqrt(accum); in Compile()
643 xla::XlaOp new_mom = mom * momentum + grad * lr * xla::Rsqrt(denominator); in Compile()
862 xla::Sqrt(accum_update + epsilon) * xla::Rsqrt(accum + epsilon) * grad; in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 94 std::unique_ptr<HloInstruction> Rsqrt( in Rsqrt() function in xla::__anon406107c40111::BatchNormExpanderVisitor
281 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormTraining()
389 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormInference()
507 add(Rsqrt(add_binary(activation_shape, HloOpcode::kAdd, in HandleBatchNormGrad()
511 auto rsqrt_var_add_epsilon = add(Rsqrt( in HandleBatchNormGrad()
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_Rsqrt.pbtxt | 2 graph_op_name: "Rsqrt"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Rsqrt.pbtxt | 2 graph_op_name: "Rsqrt"
|
| /external/shaderc/spirv-headers/include/spirv/unified1/ |
| D | OpenCL.std.h | 91 Rsqrt = 56, enumerator
|
| /external/swiftshader/third_party/SPIRV-Headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/shaderc/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/swiftshader/third_party/SPIRV-Headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/deqp-deps/SPIRV-Headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/shaderc/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/swiftshader/third_party/SPIRV-Headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/deqp-deps/SPIRV-Headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/deqp-deps/SPIRV-Headers/include/spirv/unified1/ |
| D | OpenCL.std.h | 91 Rsqrt = 56, enumerator
|
| /external/swiftshader/third_party/SPIRV-Headers/include/spirv/unified1/ |
| D | OpenCL.std.h | 91 Rsqrt = 56, enumerator
|
| /external/deqp-deps/SPIRV-Headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/shaderc/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/spirv-llvm/lib/SPIRV/libSPIRV/ |
| D | OpenCL.std.h | 92 Rsqrt = 56, enumerator
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | elementwise_test.cc | 108 TEST(ElementWise, Rsqrt) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | exhaustive_op_test.cc | 543 XLA_TEST_P(ExhaustiveOpTest, Rsqrt) { in XLA_TEST_P() argument
545 Rsqrt, +[](float x) { return 1 / std::sqrt(x); }); in XLA_TEST_P()
|