Searched refs:axis_node (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | pack_builder.cc | 47 auto* axis_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
49 AddInput(TensorID(axis_node->GetID(), 0)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms.cc | 235 NodeDef axis_node = concat_node_match.inputs[2].node; in FuseBatchNormWithConvConcat() local
236 CHECK_EQ("Const", axis_node.op()); in FuseBatchNormWithConvConcat()
237 Tensor axis = GetNodeTensorAttr(axis_node, "value"); in FuseBatchNormWithConvConcat()
|
| D | sparsify_gather_test.cc | 62 NodeDef* axis_node = in MakeGather() local
66 SetNodeTensorAttr<int32>("value", axis_t, axis_node); in MakeGather()
67 CreateNode(name, "GatherV2", {params, indices, axis_node}, graph_def); in MakeGather()
|
| D | sparsify_gather.cc | 300 const NodeDef& axis_node = match.inputs[2].node; in SparsifyGatherInternal() local
303 TF_RETURN_IF_ERROR(GetNodeAttr(axis_node, "value", &axis_t)); in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 424 auto* axis_node = context.graph_view->GetNode( in ReduceTransposerKeepDims() local
426 ASSERT_NE(axis_node, nullptr); in ReduceTransposerKeepDims()
427 ASSERT_EQ(axis_node->NumRegularFanins(), 1); in ReduceTransposerKeepDims()
428 VerifyRegularFaninMatch(axis_node, 0, "axis", 0); in ReduceTransposerKeepDims()
491 auto* axis_node = context.graph_view->GetNode( in ReduceTransposerValidAxisNode() local
493 ASSERT_NE(axis_node, nullptr); in ReduceTransposerValidAxisNode()
494 ASSERT_EQ(axis_node->NumRegularFanins(), 1); in ReduceTransposerValidAxisNode()
495 VerifyRegularFaninMatch(axis_node, 0, "axis", 0); in ReduceTransposerValidAxisNode()
2969 auto* axis_node = context.graph_view->GetNode( in TEST_F() local
2971 ASSERT_NE(axis_node, nullptr); in TEST_F()
[all …]
|
| D | constant_folding.cc | 2444 NodeDef* axis_node = optimized_graph->add_node(); in SimplifyPack() local
2445 *axis_node = std::move(new_node); in SimplifyPack()
2446 axis_node->set_name(axis_node_name); in SimplifyPack()
2447 node_map_->AddNode(axis_node->name(), axis_node); in SimplifyPack()
2451 axis_node->add_input(ctrl_dep); in SimplifyPack()
2452 axis_node->set_device(node->device()); in SimplifyPack()
2453 node_map_->AddOutput(NodeName(node->input(0)), axis_node->name()); in SimplifyPack()
2462 node->add_input(axis_node->name()); in SimplifyPack()
2463 node_map_->AddOutput(axis_node->name(), node->name()); in SimplifyPack()
|
| D | generic_layout_optimizer_transposer.cc | 1357 int axis_node = 0; in TransposeNode() local
1361 axis_node = n_attr->i(); in TransposeNode()
1368 UpdateFaninEdgesWithOp(context, {axis_node}, node, kOpDataFormatDimMap)); in TransposeNode()
1510 auto* axis_node = regular_fanin_1.node_view(); in IsReduceAxisSupported() local
1511 if (!IsConstant(*axis_node->node())) { in IsReduceAxisSupported()
1514 const auto* value_attr = axis_node->GetAttr(kAttrValue); in IsReduceAxisSupported()
|
| D | layout_optimizer.cc | 1963 auto axis_node = node_map_->GetNode(node_->input(1)); in IsAlongAxis() local
1964 if (!IsConstant(*axis_node)) { in IsAlongAxis()
1967 if (HasAttribute(*axis_node, "value").ok()) { in IsAlongAxis()
1969 auto success = tensor.FromProto(axis_node->attr().at({"value"}).tensor()); in IsAlongAxis()
|