Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Delementwise_test.cc40 OperationType op_type = OperationType::ABS; in TEST() local
52 OperationType op_type = OperationType::COS; in TEST() local
64 OperationType op_type = OperationType::DIV; in TEST() local
78 OperationType op_type = OperationType::HARD_SWISH; in TEST() local
92 OperationType op_type = OperationType::LOG; in TEST() local
104 OperationType op_type = OperationType::POW; in TEST() local
118 OperationType op_type = OperationType::RSQRT; in TEST() local
130 OperationType op_type = OperationType::SIGMOID; in TEST() local
142 OperationType op_type = OperationType::SIN; in TEST() local
154 OperationType op_type = OperationType::SQRT; 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/experimental/delegates/hexagon/builders/
Dactivation_builder.h28 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type) in ActivationOpBuilder()
30 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type, in ActivationOpBuilder()
Dresize_bilinear_builder.h26 explicit ResizeBilinearOpBuilder(GraphBuilder* graph_builder, int op_type) in ResizeBilinearOpBuilder()
Dsplit_builder.h28 explicit SplitOpBuilder(GraphBuilder* graph_builder, int op_type) in SplitOpBuilder()
Dpad_builder.h28 explicit PadOpBuilder(GraphBuilder* graph_builder, int op_type) in PadOpBuilder()
Dl2_normalization_builder.h28 explicit L2NormalizationOpBuilder(GraphBuilder* graph_builder, int op_type) in L2NormalizationOpBuilder()
Dtranspose_builder.h26 explicit TransposeOpBuilder(GraphBuilder* graph_builder, int op_type) in TransposeOpBuilder()
Darg_min_max_builder.h26 explicit ArgMinMaxOpBuilder(GraphBuilder* graph_builder, int op_type) in ArgMinMaxOpBuilder()
Dneg_op_builder.h26 explicit NegOpBuilder(GraphBuilder* graph_builder, int op_type) in NegOpBuilder()
Dreshape_builder.h28 explicit ReshapeOpBuilder(GraphBuilder* graph_builder, int op_type) in ReshapeOpBuilder()
Dresize_nearest_neighbor_builder.h29 int op_type) in ResizeNearestNeighborOpBuilder()
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py265 op_type, argument
300 op_type, argument
420 op_type, argument
510 op_type, argument
566 def _func_graph_id_from_func_name(self, op_type): argument
610 def _should_dump_tensor(self, op_type, dtype): argument
Dcheck_numerics_callback.py124 op_type, argument
235 op_type, argument
297 op_type, argument
Ddebug_events_monitors.py89 op_type, argument
112 def op_type(self): member in InfNanAlert
155 op_type, argument
199 op_type, argument
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Delementwise.cc32 OperationType op_type) { in GetElementwiseWithTwoInputsCode()
95 OperationType op_type) { in ElementwiseWithTwoInputs()
135 int id, ValueId input_id, ValueId output_id, OperationType op_type) { in ElementwiseWithOneInput()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Delementwise.cc99 const OperationType& op_type) { in CreateElementwiseOneInput()
185 const OperationDef& definition, const OperationType& op_type, in CreateElementwiseTwoInput()
193 const OperationType& op_type) { in CreateElementwiseTwoInput()
Delementwise.h33 const OperationType& op_type) in ElementwiseOneInput()
62 const OperationType& op_type, in ElementwiseTwoInput()
/external/tensorflow/tensorflow/python/eager/
Dtape.py144 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
152 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
160 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/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/ops/
Dcond_v2.py567 def _make_output_composite_tensors_match(op_type, branch_graphs): argument
617 def _make_indexed_slices_indices_types_match(op_type, branch_graphs): argument
758 def _check_same_outputs(op_type, graphs): argument
812 def verify_captures(op_type, branch_graphs): argument

123456