Home
last modified time | relevance | path

Searched defs:op_type (Results 1 – 25 of 173) 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::HARD_SWISH; in TEST() local
121 OperationType op_type = OperationType::LOG; in TEST() local
133 OperationType op_type = OperationType::NEG; in TEST() local
145 OperationType op_type = OperationType::RSQRT; in TEST() local
157 OperationType op_type = OperationType::SIGMOID; 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/core/grappler/optimizers/data/vectorization/
Dvectorizer_registry.h48 VectorizerRegistration(const string& op_type, in VectorizerRegistration()
56 #define REGISTER_VECTORIZER(op_type, vectorizer) \ argument
59 #define REGISTER_VECTORIZER_UNIQ_HELPER(ctr, op_type, vectorizer) \ argument
62 #define REGISTER_VECTORIZER_UNIQ(ctr, op_type, vectorizer) \ argument
Dvectorizer_registry.cc28 Vectorizer* VectorizerRegistry::Get(const string& op_type) { in Get()
36 void VectorizerRegistry::Register(const string& op_type, in Register()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise.cc29 const OperationType& op_type, in GetOneInputCode()
104 std::string GetTwoInputCode(const OperationType& op_type, in GetTwoInputCode()
185 const OperationDef& definition, const OperationType& op_type, in CreateElementwiseOneRuntimeOneScalar()
204 const OperationType& op_type, in CreateElementwiseTwoInput()
235 const OperationType& op_type, in CreateElementwiseTwoInput()
270 const OperationType& op_type) { in CreateElementwiseOneInput()
280 const OperationType& op_type, in CreateElementwise()
304 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.py61 def is_op_type_function(op_type): argument
292 op_type, argument
336 op_type, argument
456 op_type, argument
548 op_type, argument
621 def _func_graph_id_from_func_name(self, op_type): argument
665 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()
Dcast_builder.h32 explicit CastOpBuilder(GraphBuilder* graph_builder, int op_type) in CastOpBuilder()
Dpad_builder.h28 explicit PadOpBuilder(GraphBuilder* graph_builder, int op_type) in PadOpBuilder()
Dtranspose_builder.h26 explicit TransposeOpBuilder(GraphBuilder* graph_builder, int op_type) in TransposeOpBuilder()
Dneg_op_builder.h26 explicit NegOpBuilder(GraphBuilder* graph_builder, int op_type) in NegOpBuilder()
Dresize_nearest_neighbor_builder.h29 int op_type) in ResizeNearestNeighborOpBuilder()
Dstrided_slice_builder.h26 explicit StridedSliceOpBuilder(GraphBuilder* graph_builder, int op_type) in StridedSliceOpBuilder()
/external/tensorflow/tensorflow/core/profiler/utils/
Dtf_op_utils.cc50 bool IsTfOpType(absl::string_view op_type) { in IsTfOpType()
55 bool IsJaxOpType(absl::string_view op_type) { in IsJaxOpType()
60 bool IsJaxOpNameAndType(absl::string_view op_name, absl::string_view op_type) { in IsJaxOpNameAndType()
/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/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/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

1234567