Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Delementwise_test.cc43 OperationType op_type = OperationType::ABS; in TEST() local
55 OperationType op_type = OperationType::COS; in TEST() local
67 OperationType op_type = OperationType::COPY; in TEST() local
78 OperationType op_type = OperationType::ELU; in TEST() local
93 OperationType op_type = OperationType::EXP; in TEST() local
109 OperationType op_type = OperationType::FLOOR; in TEST() local
123 OperationType op_type = OperationType::HARD_SWISH; in TEST() local
137 OperationType op_type = OperationType::LOG; in TEST() local
149 OperationType op_type = OperationType::NEG; in TEST() local
161 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.cc64 OperationType op_type = OperationType::PRELU; variable
89 OperationType op_type = OperationType::PRELU; variable
114 OperationType op_type = OperationType::PRELU; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise.cc31 const OperationType& op_type, in GetOneInputCode()
145 std::string GetTwoInputCode(const OperationType& op_type, in GetTwoInputCode()
232 const OperationDef& definition, const OperationType& op_type, in CreateElementwiseOneRuntimeOneScalar()
250 const OperationType& op_type, in CreateElementwiseTwoInput()
276 const OperationType& op_type, in CreateElementwiseTwoInput()
307 const OperationType& op_type, in CreateElementwiseDesc()
334 const OperationType& op_type) { in CreateElementwiseOneInput()
343 const OperationType& op_type, in CreateElementwise()
350 const OperationType& op_type, in CreateElementwiseTwoInput()
404 const OperationType& op_type, const BHWC& input_shape, in CreateElementwiseOneInputWithBroadcast()
[all …]
Dreduce.cc59 std::string MakeOp(OperationType op_type, const std::string& a, in MakeOp()
128 Reduce::Reduce(const std::map<Axis, int>& axis_to_reduce, OperationType op_type, in Reduce()
180 OperationType op_type) { in GetReduceKernelCode()
526 OperationType op_type, const OperationDef& definition, in CreateReduce()
533 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
/external/tensorflow/tensorflow/core/profiler/lib/
Dtraceme_encode.h136 absl::string_view op_name, absl::string_view op_type) { in TraceMeOp()
141 const char* op_type) { in TraceMeOp()
146 std::string&& op_name, absl::string_view op_type) { in TraceMeOp()
153 absl::string_view op_name, absl::string_view op_type) { in TraceMeOpOverride()
158 const char* op_name, const char* op_type) { in TraceMeOpOverride()
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py55 def is_op_type_function(op_type): argument
286 op_type, argument
330 op_type, argument
447 op_type, argument
539 op_type, argument
612 def _func_graph_id_from_func_name(self, op_type): argument
656 def _should_dump_tensor(self, op_type, dtype): argument
Ddebug_events_monitors.py85 op_type, argument
108 def op_type(self): member in InfNanAlert
151 op_type, argument
195 op_type, argument
Dcheck_numerics_callback.py125 op_type, argument
236 op_type, argument
300 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()
Dreduce_builder.h28 explicit ReduceOpBuilder(GraphBuilder* graph_builder, int op_type) in ReduceOpBuilder()
Dpad_builder.h28 explicit PadOpBuilder(GraphBuilder* graph_builder, int op_type) in PadOpBuilder()
Dneg_op_builder.h26 explicit NegOpBuilder(GraphBuilder* graph_builder, int op_type) in NegOpBuilder()
Dsplit_builder.h28 explicit SplitOpBuilder(GraphBuilder* graph_builder, int op_type) in SplitOpBuilder()
Dcast_builder.h32 explicit CastOpBuilder(GraphBuilder* graph_builder, int op_type) in CastOpBuilder()
/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/XNNPACK/test/
Dvbinaryc-microkernel-tester.h85 void Test(xnn_f16_vbinary_ukernel_function vbinaryc, OpType op_type) const { in Test()
149 …void Test(xnn_f16_vbinary_minmax_ukernel_function vbinaryc_minmax, OpType op_type, xnn_init_f16_mi… in Test()
302 void Test(xnn_f32_vbinary_relu_ukernel_function vbinaryc_relu, OpType op_type) const { in Test()
372 …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()
153 …void Test(xnn_f16_vbinary_minmax_ukernel_function vbinary_minmax, OpType op_type, xnn_init_f16_min… in Test()
299 void Test(xnn_f32_vbinary_relu_ukernel_function vbinary_relu, OpType op_type) const { in Test()
365 …void Test(xnn_f32_vbinary_minmax_ukernel_function vbinary_minmax, OpType op_type, xnn_init_f32_min… in Test()
/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.py177 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
185 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
193 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
/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/core/framework/
Dop_kernel_test_base.h56 NodeDef CreateNodeDef(const string& op_type, in CreateNodeDef()
112 void ExpectFailure(const string& op_type, const DeviceType& device_type, in ExpectFailure()

12345678