| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | gpu_op_rsqrt.cc | 21 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Rsqrt, DT_HALF);
22 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Rsqrt, DT_FLOAT);
23 GENERATE_AND_REGISTER_UNARY_GPU_KERNEL(Rsqrt, DT_DOUBLE);
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Rsqrt.pbtxt | 2 name: "Rsqrt"
26 name: "Rsqrt"
51 name: "Rsqrt"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Rsqrt.pbtxt | 2 name: "Rsqrt"
26 name: "Rsqrt"
51 name: "Rsqrt"
|
| /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/compiler/xla/tests/ |
| D | exhaustive_unary_test_complex.cc | 114 UNARY_TEST_COMPLEX_64(Rsqrt, {
115 Run(Rsqrt, [](complex64 x) { in __anonf9531efa0502()
200 UNARY_TEST_COMPLEX_128(Rsqrt, {
215 Rsqrt,
|
| D | exhaustive_unary_test_f64.cc | 75 UNARY_TEST_FLOAT_64(Rsqrt, {
77 Rsqrt, +[](double x) { return 1 / std::sqrt(x); }); in __anon5d9703520302()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unary_ops_composition.cc | 242 REGISTER_COMPUTE_FN(Rsqrt); in UnaryOpsCompositionSupport()
281 REGISTER_COMPUTE_FN_HELPER(Rsqrt, functor::rsqrt<T>);
310 REGISTER_COMPUTE_FN(Rsqrt); in UnaryOpsCompositionSupport()
338 REGISTER_COMPUTE_FN_HELPER(Rsqrt, functor::rsqrt<T>);
372 REGISTER_COMPUTE_FN(Rsqrt); in UnaryOpsCompositionSupport()
410 REGISTER_COMPUTE_FN_HELPER(Rsqrt, functor::rsqrt<T>);
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_Rsqrt.pbtxt | 2 graph_op_name: "Rsqrt"
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd.cc | 369 auto c_temp = Rsqrt(one + Square(t)); in MakeJacobi()
376 auto rnorm = Rsqrt(Square(c) + Square(s)); in MakeJacobi()
426 XlaOp tmp = Rsqrt(one + Square(u)); in GetOneSidedJacobiRotation()
514 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in OneSidedJacobiUpdate()
524 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in OneSidedJacobiUpdate()
540 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in OneSidedJacobiUpdate()
550 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in OneSidedJacobiUpdate()
762 Rsqrt(Reduce(Square(result.v), ScalarLike(d, 0.0), in SortBySingularValuesAndPostProcessing()
774 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/core/api_def/base_api/ |
| D | api_def_Rsqrt.pbtxt | 2 graph_op_name: "Rsqrt"
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/op_definitions/ |
| D | rsqrt.mlir.tmpl | 3 %0 = "tf.Rsqrt"(%arg0) : (tensor<*xelem_type>) -> tensor<*xoutput_type>
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 86 std::unique_ptr<HloInstruction> Rsqrt( in Rsqrt() function in xla::__anon62d847220111::BatchNormExpanderVisitor
248 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormTraining()
345 add(Rsqrt(add_binary(feature_shape, HloOpcode::kAdd, var, epsilon), in HandleBatchNormInference()
456 add(Rsqrt(add_binary(activation_shape, HloOpcode::kAdd, in HandleBatchNormGrad()
460 auto rsqrt_var_add_epsilon = add(Rsqrt( in HandleBatchNormGrad()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unary_ops.cc | 87 XLAJIT_MAKE_UNARY(Rsqrt, xla::Rsqrt(x));
|
| /external/deqp-deps/SPIRV-Headers/include/spirv/1.1/ |
| 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.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/angle/third_party/vulkan-deps/spirv-headers/src/include/spirv/1.0/ |
| 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/shaderc/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/shaderc/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /external/angle/third_party/vulkan-deps/spirv-headers/src/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/angle/third_party/vulkan-deps/spirv-headers/src/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|