Home
last modified time | relevance | path

Searched refs:ElementwiseAttributes (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_mul_to_conv.h40 void FuseConvolution2DWithMultiply(const ElementwiseAttributes& mul_attr,
47 const ElementwiseAttributes& mul_attr,
54 const ElementwiseAttributes& mul_attr,
60 void FuseFullyConnectedWithMultiply(const ElementwiseAttributes& mul_attr,
66 void FuseMultiplyWithConvolution2D(const ElementwiseAttributes& mul_attr,
73 const ElementwiseAttributes& mul_attr,
80 const ElementwiseAttributes& mul_attr,
86 void FuseMultiplyWithFullyConnected(const ElementwiseAttributes& mul_attr,
Dfuse_add_to_conv.h35 void FuseConvolution2DWithAdd(const ElementwiseAttributes& add_attr,
41 void FuseDepthwiseConvolution2DWithAdd(const ElementwiseAttributes& add_attr,
47 void FuseConvolutionTransposedWithAdd(const ElementwiseAttributes& add_attr,
53 void FuseFullyConnectedWithAdd(const ElementwiseAttributes& add_attr,
Dfuse_add_to_conv.cc37 void FuseBiasWithAddAttributes(const ElementwiseAttributes& add_attr, in FuseBiasWithAddAttributes()
65 ElementwiseAttributes add_attr = in ApplyToNodesSequence()
66 absl::any_cast<ElementwiseAttributes>(add_node.operation.attributes); in ApplyToNodesSequence()
117 void FuseConvolution2DWithAdd(const ElementwiseAttributes& add_attr, in FuseConvolution2DWithAdd()
122 void FuseDepthwiseConvolution2DWithAdd(const ElementwiseAttributes& add_attr, in FuseDepthwiseConvolution2DWithAdd()
128 void FuseConvolutionTransposedWithAdd(const ElementwiseAttributes& add_attr, in FuseConvolutionTransposedWithAdd()
133 void FuseFullyConnectedWithAdd(const ElementwiseAttributes& add_attr, in FuseFullyConnectedWithAdd()
Dfuse_mul_to_conv.cc55 ElementwiseAttributes mul_attr = in ApplyToNodesSequence()
56 absl::any_cast<ElementwiseAttributes>(mul_node.operation.attributes); in ApplyToNodesSequence()
119 ElementwiseAttributes mul_attr = in ApplyToNodesSequence()
120 absl::any_cast<ElementwiseAttributes>(mul_node.operation.attributes); in ApplyToNodesSequence()
176 void FuseConvolution2DWithMultiply(const ElementwiseAttributes& mul_attr, in FuseConvolution2DWithMultiply()
197 const ElementwiseAttributes& mul_attr, in FuseDepthwiseConvolution2DWithMultiply()
219 const ElementwiseAttributes& mul_attr, in FuseConvolutionTransposedWithMultiply()
239 void FuseFullyConnectedWithMultiply(const ElementwiseAttributes& mul_attr, in FuseFullyConnectedWithMultiply()
255 void FuseMultiplyWithConvolution2D(const ElementwiseAttributes& mul_attr, in FuseMultiplyWithConvolution2D()
273 const ElementwiseAttributes& mul_attr, in FuseMultiplyWithDepthwiseConvolution2D()
[all …]
Dfuse_mul_to_conv_test.cc58 ElementwiseAttributes mul_attr; in TEST()
99 ElementwiseAttributes mul_attr; in TEST()
152 ElementwiseAttributes mul_attr; in TEST()
173 ElementwiseAttributes mul_attr; in TEST()
195 ElementwiseAttributes mul_attr; in TEST()
216 ElementwiseAttributes mul_attr; in TEST()
236 ElementwiseAttributes mul_attr; in TEST()
257 ElementwiseAttributes mul_attr; in TEST()
279 ElementwiseAttributes mul_attr; in TEST()
300 ElementwiseAttributes mul_attr; in TEST()
Dfuse_add_to_conv_test.cc58 ElementwiseAttributes add_attr; in TEST()
101 ElementwiseAttributes add_attr; in TEST()
122 ElementwiseAttributes add_attr; in TEST()
144 ElementwiseAttributes add_attr; in TEST()
165 ElementwiseAttributes add_attr; in TEST()
Dremove_noop_test.cc49 add_node->operation.attributes = ElementwiseAttributes(); in TEST()
77 ElementwiseAttributes attr; in TEST()
104 ElementwiseAttributes attr; in TEST()
131 ElementwiseAttributes attr; in TEST()
Dmerge_padding_with.cc153 ElementwiseAttributes add_attr = in ApplyToNode()
154 absl::any_cast<ElementwiseAttributes>(add_node->operation.attributes); in ApplyToNode()
Dadd_quant_adjustments_test.cc60 ElementwiseAttributes add_attr; in TEST()
104 ElementwiseAttributes add_attr; in TEST()
Dmerge_padding_with_test.cc137 ElementwiseAttributes add_attr; in TEST()
175 ElementwiseAttributes add_attr; in TEST()
Dremove_noop.cc89 auto& attr = absl::any_cast<const ElementwiseAttributes&>( in NewRemoveSingleInputAdd()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dadd_test.cc47 ElementwiseAttributes attr; in TEST()
58 ElementwiseAttributes attr; in TEST()
83 ElementwiseAttributes attr; in TEST()
117 ElementwiseAttributes attr; in TEST()
Delementwise_test.cc248 ElementwiseAttributes attr; in TEST()
263 ElementwiseAttributes attr; in TEST()
311 ElementwiseAttributes attr; in TEST()
327 ElementwiseAttributes attr; in TEST()
375 ElementwiseAttributes attr; in TEST()
391 ElementwiseAttributes attr; in TEST()
439 ElementwiseAttributes attr; in TEST()
455 ElementwiseAttributes attr; in TEST()
503 ElementwiseAttributes attr; in TEST()
518 ElementwiseAttributes attr; in TEST()
[all …]
Dmul_test.cc44 ElementwiseAttributes attr; in TEST()
64 ElementwiseAttributes attr; in TEST()
88 ElementwiseAttributes attr; in TEST()
Dadd.cc40 absl::any_cast<const ElementwiseAttributes&>(ctx.op_attr); in GenerateCode()
Dmul.cc84 const auto& attr = absl::any_cast<const ElementwiseAttributes&>(ctx.op_attr); in GenerateMultiplyScalarCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.cc161 ElementwiseAttributes attr; in BuildLstmGate()
210 ElementwiseAttributes attr; in BuildLstmGate()
224 ElementwiseAttributes attr; in BuildLstmGate()
296 ElementwiseAttributes attr; in BuildCellStateUpdate()
307 ElementwiseAttributes attr; in BuildCellStateUpdate()
406 ElementwiseAttributes attr; in BuildOutputStateUpdate()
417 ElementwiseAttributes attr; in BuildOutputStateUpdate()
482 ElementwiseAttributes attr; in ParseLSTMAttributes()
Doperations.h512 struct ElementwiseAttributes { struct
Dmodel_builder.cc197 ElementwiseAttributes attr; in Parse()
665 ElementwiseAttributes attr; in Parse()
1111 ElementwiseAttributes attr; in Parse()
1133 ElementwiseAttributes attr; in Parse()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Delementwise.h39 const ElementwiseAttributes& attr);
Delementwise_test_util.cc588 ElementwiseAttributes attr; in MaximumWithScalarTest()
621 ElementwiseAttributes attr; in MaximumWithConstantLinearTensorTest()
653 ElementwiseAttributes attr; in MaximumWithConstantHWCTensorTest()
685 ElementwiseAttributes attr; in MaximumWithConstantHWCTensorBroadcastChannelsTest()
744 ElementwiseAttributes attr; in MinimumWithScalarTest()
863 ElementwiseAttributes attr; in SubWithScalarAtFirstPositionTest()
923 ElementwiseAttributes attr; in LessEqualTest()
952 ElementwiseAttributes attr; in GreaterTest()
981 ElementwiseAttributes attr; in GreaterEqualTest()
1010 ElementwiseAttributes attr; in EqualTest()
[all …]
Delementwise.cc281 const ElementwiseAttributes& attr) { in CreateElementwise()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc180 absl::any_cast<ElementwiseAttributes>(node.operation.attributes); in GPUOperationFromNode()
544 absl::any_cast<ElementwiseAttributes>(node.operation.attributes); in GPUOperationFromNode()