Home
last modified time | relevance | path

Searched refs:ApplyToNode (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
Dfuse_auto_input_test.cc62 EXPECT_EQ(fuse_auto_input.ApplyToNode(n3, &graph).status, in TEST()
93 EXPECT_EQ(fuse_auto_input.ApplyToNode(n2, &graph).status, in TEST()
Dfuse_auto_input.h42 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final;
Dfuse_inplace.h60 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final;
Dfuse_inplace.cc80 TransformResult RemoveUnusedInplaceUpdates::ApplyToNode(Node* node, in ApplyToNode() function in tflite::gpu::gl::RemoveUnusedInplaceUpdates
Dfuse_auto_input.cc46 TransformResult FuseAutoInput::ApplyToNode(Node* node, GraphFloat32* graph) { in ApplyToNode() function in tflite::gpu::gl::FuseAutoInput
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmake_fully_connected.cc45 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::__anone2bd68f10111::MakeFullyConnectedFromConvolution
Dglobal_pooling_to_reduce_op.cc51 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::__anon4bc9aed50111::GlobalPoolingToReduceOp
Dadd_quant_adjustments.cc37 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::AddQuantAdjustments
Dmake_padding.cc52 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::__anon1af4062a0111::MakePaddingFromZerosConcat
Dadd_bias.cc54 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::__anon89aaf7dd0111::AddBias
Dremove_noop.cc115 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::RemoveIdentityReshape
Dmerge_padding_with.cc115 TransformResult ApplyToNode(Node* node, GraphFloat32* graph) final { in ApplyToNode() function in tflite::gpu::MergePaddingWithAddOperation
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_transformer.h66 virtual TransformResult ApplyToNode(Node* node, GraphFloat32* graph) = 0;
Dmodel_transformer.cc64 auto result = transformation->ApplyToNode(node, graph_); in Apply()