Searched refs:sq_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | identify_l2_pool_test.cc | 41 auto sq_op = new TensorFlowSquareOperator; in RunIdentifyL2Pool() local
42 sq_op->inputs = {"input0"}; in RunIdentifyL2Pool()
43 sq_op->outputs = {"output"}; in RunIdentifyL2Pool()
49 avgpool_op->inputs = {sq_op->outputs[0]}; in RunIdentifyL2Pool()
62 model.operators.push_back(std::unique_ptr<Operator>(sq_op)); in RunIdentifyL2Pool()
|
| D | identify_l2_normalization_test.cc | 42 auto sq_op = new TensorFlowSquareOperator; in RunIdentifyL2Normalization() local
43 sq_op->inputs = {"input0"}; in RunIdentifyL2Normalization()
44 sq_op->outputs = {"output"}; in RunIdentifyL2Normalization()
50 sum_op->inputs = {sq_op->outputs[0]}; in RunIdentifyL2Normalization()
72 model.operators.push_back(std::unique_ptr<Operator>(sq_op)); in RunIdentifyL2Normalization()
92 model.operators.push_back(std::unique_ptr<Operator>(sq_op)); in RunIdentifyL2Normalization()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_patterns.td | 315 (TFL_SquareOp:$sq_op $x),
320 [(L2NormValidReduceIndex $sq_op, $axis)]>;
329 (TFL_SquareOp:$sq_op $x),
335 [(L2NormValidReduceIndex $sq_op, $axis),
343 (TFL_SquareOp:$sq_op $x),
349 [(L2NormValidReduceIndex $sq_op, $axis),
|
| D | optimize.cc | 66 bool L2NormalizeReduceAxis(Value sq_op, DenseElementsAttr axis) { in L2NormalizeReduceAxis() argument
67 if (sq_op.getType().cast<ShapedType>().getRank() - 1 == in L2NormalizeReduceAxis()
72 if (sq_op.getType().cast<ShapedType>().getRank() != axis.getNumElements()) { in L2NormalizeReduceAxis()
75 auto shape = sq_op.getType().cast<ShapedType>(); in L2NormalizeReduceAxis()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 772 sq_op = sq.get_concrete_function(t)
773 self.assertEqual(sq_op.output_shapes, tensor_shape.TensorShape([2, 2]))
774 out = sq_op(t)
815 sq_op = sq.get_concrete_function(
817 self.assertEqual([None, None], sq_op.output_shapes.as_list())
820 out1 = sq_op(t1)
824 out2 = sq_op(t2)
840 sq_op = sq.get_concrete_function(
845 self.assertEqual([None, None], sq_op.output_shapes.as_list())
849 out = sq_op(first_mat=t1, second_mat=t2)
[all …]
|