| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | cwise_op_rsqrt_benchmark.cc | 22 %0 = "tf.Rsqrt"(%arg0): (tensor<?xf32>) -> tensor<?xf32>
32 BM_TFMlir(Rsqrt, mlir_input, "rsqrt_1d", 1, f32, 1.0, 0.0, /* num_threads */ 0)
38 BM_TFMlirVectorized(Rsqrt, mlir_input, "rsqrt_1d", 1, f32, 1.0, 0.0,
45 BM_EigenScalar(Rsqrt, EXPR_BUILDER, 1, f32, 1.0, 0.0, /* num_threads */ 0)
51 BM_EigenVectorized(Rsqrt, EXPR_BUILDER, 1, f32, 1.0, 0.0, /* num_threads */ 0)
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | auto-fusion.mlir | 1 …-executor-to-tfrt-pipeline="enable-native-ops=false auto-fusion-oplist=tf.Rsqrt,tf.Tanh auto-fusio…
16 %0 = "tf.Rsqrt"(%arg0) {T = f32, device="/device:CPU:0"} : (tensor<?xf32>) -> tensor<?xf32>
26 // CHECK: %[[RES:.*]] = "tf.Rsqrt"(%[[ARG0]])
34 // CHECK-NOT: Rsqrt
36 %0 = "tf.Rsqrt"(%arg0) {T = f32, device="/device:CPU:0"} : (tensor<?xf32>) -> tensor<?xf32>
47 // CHECK: %[[RES0:.*]] = "tf.Rsqrt"(%[[ARG0]])
56 %0 = "tf.Rsqrt"(%arg0) {T = f32, device="/device:CPU:0"} : (tensor<?xf32>) -> tensor<?xf32>
65 // CHECK: tf.Rsqrt
|
| D | outline-cpurt-cluster.mlir | 13 %1 = "tf.Rsqrt"(%arg0) : (tensor<?xf32>) -> tensor<?xf32>
26 // CHECK: %[[RET:.*]] = "tf.Rsqrt"(%arg0)
|
| /external/armnn/src/profiling/test/ |
| D | FileOnlyProfilingDecoratorTests.cpp | 70 ElementwiseUnaryDescriptor descriptor(UnaryOperation::Rsqrt);
71 IConnectableLayer* Rsqrt = net->AddElementwiseUnaryLayer(descriptor, "Rsqrt"); variable
75 input->GetOutputSlot(0).Connect(Rsqrt->GetInputSlot(0));
76 Rsqrt->GetOutputSlot(0).Connect(output->GetInputSlot(0));
79 Rsqrt->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 4, 4 }, DataType::Float32));
194 ElementwiseUnaryDescriptor descriptor(UnaryOperation::Rsqrt);
195 IConnectableLayer* Rsqrt = net->AddElementwiseUnaryLayer(descriptor, "Rsqrt"); variable
199 input->GetOutputSlot(0).Connect(Rsqrt->GetInputSlot(0));
200 Rsqrt->GetOutputSlot(0).Connect(output->GetInputSlot(0));
203 Rsqrt->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 4, 4 }, DataType::Float32));
|
| /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);
|
| D | cpu_op_rsqrt.cc | 21 GENERATE_AND_REGISTER_UNARY_CPU_KERNEL(Rsqrt, DT_HALF);
22 GENERATE_AND_REGISTER_UNARY_CPU_KERNEL(Rsqrt, DT_FLOAT);
23 GENERATE_AND_REGISTER_UNARY_CPU_KERNEL(Rsqrt, DT_DOUBLE);
|
| /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/ops/compat/ops_history_v2/ |
| D | Rsqrt.pbtxt | 2 name: "Rsqrt"
26 name: "Rsqrt"
51 name: "Rsqrt"
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/ |
| D | tf_jitrt_stream_analysis.mlir | 5 %0 = "tf.Rsqrt"(%arg0): (tensor<512xf32>) -> tensor<512xf32>
12 %0 = "tf.Rsqrt"(%arg0): (tensor<512x512xf32>) -> tensor<512x512xf32>
19 %0 = "tf.Rsqrt"(%arg0): (tensor<?x512xf32>) -> tensor<?x512xf32>
20 %1 = "tf.Rsqrt"(%0): (tensor<?x512xf32>) -> tensor<?x512xf32>
21 %2 = "tf.Rsqrt"(%1): (tensor<?x512xf32>) -> tensor<?x512xf32>
22 %3 = "tf.Rsqrt"(%2): (tensor<?x512xf32>) -> tensor<?x512xf32>
23 %4 = "tf.Rsqrt"(%3): (tensor<?x512xf32>) -> tensor<?x512xf32>
|
| D | tf_cpurt_clustering.mlir | 130 // CHECK: tf.Rsqrt
133 %2 = "tf.Rsqrt"(%1): (tensor<?x?xf32>) -> tensor<?x?xf32>
138 // CHECK: tf.Rsqrt
141 %5 = "tf.Rsqrt"(%4): (tensor<?x?xf32>) -> tensor<?x?xf32>
|
| D | tf_jitrt_clustering.mlir | 132 // CHECK: tf.Rsqrt
135 %2 = "tf.Rsqrt"(%1): (tensor<?x?xf32>) -> tensor<?x?xf32>
140 // CHECK: tf.Rsqrt
143 %5 = "tf.Rsqrt"(%4): (tensor<?x?xf32>) -> tensor<?x?xf32>
|
| /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 __anone74288180502()
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 __anon72b28d700302()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | RsqrtTestImpl.cpp | 33 armnn::UnaryOperation::Rsqrt, in Rsqrt2dTest()
64 armnn::UnaryOperation::Rsqrt, in Rsqrt3dTest()
93 armnn::UnaryOperation::Rsqrt, in RsqrtZeroTest()
122 armnn::UnaryOperation::Rsqrt, in RsqrtNegativeTest()
|
| /external/armnn/src/armnn/layers/ |
| D | RsqrtLayer.cpp | 18 : Layer(1, 1, LayerType::Rsqrt, name) in RsqrtLayer()
25 descriptor.m_Parameters.m_Operation = UnaryOperation::Rsqrt; in CreateWorkload()
|
| /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/compiler/xla/client/lib/ |
| D | svd.cc | 370 auto c_temp = Rsqrt(one + Square(t)); in MakeJacobi()
377 auto rnorm = Rsqrt(Square(c) + Square(s)); in MakeJacobi()
427 XlaOp tmp = Rsqrt(one + Square(u)); in GetOneSidedJacobiRotation()
515 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in OneSidedJacobiUpdate()
525 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in OneSidedJacobiUpdate()
541 Rsqrt(Reduce(Square(slice_p_new), pq_zero, in OneSidedJacobiUpdate()
551 Rsqrt(Reduce(Square(slice_q_new), pq_zero, in OneSidedJacobiUpdate()
777 Rsqrt(Reduce(Square(result.v), ScalarLike(d, 0.0), in SortBySingularValuesAndPostProcessing()
789 Rsqrt(Reduce(Square(result.u), ScalarLike(d, 0.0), in SortBySingularValuesAndPostProcessing()
|
| /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/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 | 85 std::unique_ptr<HloInstruction> Rsqrt( in Rsqrt() function in xla::__anon854f86800111::BatchNormExpanderVisitor
249 auto rsqrt_var_add_epsilon = add(Rsqrt(var_add_epsilon, add)); in HandleBatchNormTraining()
348 add(Rsqrt(add_binary(feature_shape, HloOpcode::kAdd, var, epsilon), in HandleBatchNormInference()
461 add(Rsqrt(add_binary(activation_shape, HloOpcode::kAdd, in HandleBatchNormGrad()
465 auto rsqrt_var_add_epsilon = add(Rsqrt( in HandleBatchNormGrad()
|
| /external/armnn/src/armnnDeserializer/test/ |
| D | DeserializeRsqrt.cpp | 52 layerType: "Rsqrt", in RsqrtFixture()
|
| /external/armnn/src/backends/tosaCommon/operatorMappings/ |
| D | ElementwiseUnaryOperator.cpp | 33 case UnaryOperation::Rsqrt: in ConvertElementwiseUnaryOperator()
|
| /external/angle/third_party/vulkan-deps/spirv-headers/src/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|