Home
last modified time | relevance | path

Searched defs:op_type (Results 1 – 25 of 165) sorted by relevance

1234567

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Delementwise_test.cc41 OperationType op_type = OperationType::ABS; in TEST() local
53 OperationType op_type = OperationType::COS; in TEST() local
65 OperationType op_type = OperationType::COPY; in TEST() local
76 OperationType op_type = OperationType::ELU; in TEST() local
91 OperationType op_type = OperationType::EXP; in TEST() local
107 OperationType op_type = OperationType::FLOOR; in TEST() local
121 OperationType op_type = OperationType::HARD_SWISH; in TEST() local
135 OperationType op_type = OperationType::LOG; in TEST() local
147 OperationType op_type = OperationType::NEG; in TEST() local
159 OperationType op_type = OperationType::RSQRT; in TEST() local
[all …]
Drelu_test.cc46 OperationType op_type = OperationType::RELU; in TEST_F() local
59 OperationType op_type = OperationType::RELU; in TEST_F() local
72 OperationType op_type = OperationType::RELU; in TEST_F() local
85 OperationType op_type = OperationType::RELU; in TEST_F() local
Dprelu_test.cc89 OperationType op_type = OperationType::PRELU; variable
115 OperationType op_type = OperationType::PRELU; variable
141 OperationType op_type = OperationType::PRELU; variable
167 OperationType op_type = OperationType::PRELU; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise.cc29 const OperationType& op_type, in GetOneInputCode()
107 std::string GetTwoInputCode(const OperationType& op_type, in GetTwoInputCode()
194 const OperationDef& definition, const OperationType& op_type, in CreateElementwiseOneRuntimeOneScalar()
213 const OperationType& op_type, in CreateElementwiseTwoInput()
248 const OperationType& op_type, in CreateElementwiseTwoInput()
287 const OperationType& op_type) { in CreateElementwiseOneInput()
297 const OperationType& op_type, in CreateElementwise()
321 const OperationType& op_type, in CreateElementwiseTwoInput()
Dreduce.cc57 std::string MakeOp(OperationType op_type, const std::string& a, in MakeOp()
112 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce()
163 OperationType op_type) { in GetReduceKernelCode()
476 OperationType op_type, const OperationDef& definition, in CreateReduce()
483 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
/external/tensorflow/tensorflow/core/profiler/lib/
Dtraceme_encode.h133 absl::string_view op_name, absl::string_view op_type) { in TraceMeOp()
138 const char* op_type) { in TraceMeOp()
143 std::string&& op_name, absl::string_view op_type) { in TraceMeOp()
150 absl::string_view op_name, absl::string_view op_type) { in TraceMeOpOverride()
155 const char* op_name, const char* op_type) { in TraceMeOpOverride()
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py60 def is_op_type_function(op_type): argument
291 op_type, argument
335 op_type, argument
452 op_type, argument
544 op_type, argument
617 def _func_graph_id_from_func_name(self, op_type): argument
661 def _should_dump_tensor(self, op_type, dtype): argument
Ddebug_events_monitors.py89 op_type, argument
112 def op_type(self): member in InfNanAlert
155 op_type, argument
199 op_type, argument
Dcheck_numerics_callback.py129 op_type, argument
240 op_type, argument
304 op_type, argument
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dquantize_builder.h26 explicit QuantizeOpBuilder(GraphBuilder* graph_builder, int op_type) in QuantizeOpBuilder()
28 explicit QuantizeOpBuilder(GraphBuilder* graph_builder, int op_type, in QuantizeOpBuilder()
Dactivation_builder.h28 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type) in ActivationOpBuilder()
30 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type, in ActivationOpBuilder()
Dmatmul_builder.h30 int op_type) in MatMulWithConstWeightsOpBuilder()
49 explicit MatMulOpBuilder(GraphBuilder* graph_builder, int op_type) in MatMulOpBuilder()
Dsquared_difference.cc23 explicit SquaredDifferenceOpBuilder(GraphBuilder* graph_builder, int op_type) in SquaredDifferenceOpBuilder()
98 int op_type) { in CreateSquaredDifferenceOpBuilder()
Dpad_builder.h28 explicit PadOpBuilder(GraphBuilder* graph_builder, int op_type) in PadOpBuilder()
Dcast_builder.h32 explicit CastOpBuilder(GraphBuilder* graph_builder, int op_type) in CastOpBuilder()
Darg_min_max_builder.h26 explicit ArgMinMaxOpBuilder(GraphBuilder* graph_builder, int op_type) in ArgMinMaxOpBuilder()
Dstrided_slice_builder.h26 explicit StridedSliceOpBuilder(GraphBuilder* graph_builder, int op_type) in StridedSliceOpBuilder()
Dl2_normalization_builder.h28 explicit L2NormalizationOpBuilder(GraphBuilder* graph_builder, int op_type) in L2NormalizationOpBuilder()
/external/tensorflow/tensorflow/core/profiler/utils/
Dtf_op_utils.cc55 absl::string_view op_type = op_name; in DeriveOpType() local
79 bool IsTfOpType(absl::string_view op_type) { in IsTfOpType()
84 bool IsJaxOpType(absl::string_view op_type) { in IsJaxOpType()
89 bool IsJaxOpNameAndType(absl::string_view op_name, absl::string_view op_type) { in IsJaxOpNameAndType()
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc54 void MakeOp(const OpType op_type, const Tensor& x_data, const Input& y_shape, in MakeOp()
79 void TestResizedShapeForType(const OpType op_type, const bool align_corners, in TestResizedShapeForType()
94 void TestResizedShape(OpType op_type) { in TestResizedShape()
111 void TestResizeToSmallerAndAlign(const OpType op_type, in TestResizeToSmallerAndAlign()
126 void TestResizeToLargerAndAlign(const OpType op_type, in TestResizeToLargerAndAlign()
141 void TestResize(OpType op_type) { in TestResize()
/external/tensorflow/tensorflow/python/eager/
Dtape.py181 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
189 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
197 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
/external/XNNPACK/test/
Dvbinaryc-microkernel-tester.h85 void Test(xnn_f16_vbinary_ukernel_function vbinaryc, OpType op_type) const { in Test()
150 …void Test(xnn_f16_vbinary_minmax_ukernel_function vbinaryc_minmax, OpType op_type, xnn_init_f16_mi… in Test()
304 void Test(xnn_f32_vbinary_relu_ukernel_function vbinaryc_relu, OpType op_type) const { in Test()
374 …void Test(xnn_f32_vbinary_minmax_ukernel_function vbinaryc_minmax, OpType op_type, xnn_init_f32_mi… in Test()
Dvbinary-microkernel-tester.h92 void Test(xnn_f16_vbinary_ukernel_function vbinary, OpType op_type) const { in Test()
154 …void Test(xnn_f16_vbinary_minmax_ukernel_function vbinary_minmax, OpType op_type, xnn_init_f16_min… in Test()
301 void Test(xnn_f32_vbinary_relu_ukernel_function vbinary_relu, OpType op_type) const { in Test()
367 …void Test(xnn_f32_vbinary_minmax_ukernel_function vbinary_minmax, OpType op_type, xnn_init_f32_min… in Test()
/external/google-breakpad/src/third_party/libdisasm/
Dia32_operand.c232 static size_t decode_operand_size( unsigned int op_type, x86_insn_t *insn, in decode_operand_size()
383 unsigned int addr_meth, op_type, op_size, gen_regs; in ia32_decode_operand() local
/external/tensorflow/tensorflow/python/ops/
Dcond_v2.py581 def _make_output_composite_tensors_match(op_type, branch_graphs): argument
631 def _make_indexed_slices_indices_types_match(op_type, branch_graphs): argument
763 def _check_same_outputs(op_type, graphs): argument
851 def verify_captures(op_type, branch_graphs): argument

1234567