Searched refs:axis_node (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms.cc | 236 NodeDef axis_node = concat_node_match.inputs[2].node; in FuseBatchNormWithConvConcat() local
237 CHECK_EQ("Const", axis_node.op()); in FuseBatchNormWithConvConcat()
238 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 | 301 const NodeDef& axis_node = match.inputs[2].node; in SparsifyGatherInternal() local
304 TF_RETURN_IF_ERROR(GetNodeAttr(axis_node, "value", &axis_t)); in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2298 NodeDef* axis_node = optimized_graph->add_node(); in SimplifyPack() local
2299 axis_node->set_name(OptimizedNodeName(*node, "_const_axis")); in SimplifyPack()
2303 !CreateNodeDef(axis_node->name(), TensorValue(&axis_t), axis_node) in SimplifyPack()
2310 axis_node->add_input(ctrl_dep); in SimplifyPack()
2311 axis_node->set_device(node->device()); in SimplifyPack()
2320 node->add_input(axis_node->name()); in SimplifyPack()
|
| D | layout_optimizer.cc | 1904 auto axis_node = node_map_->GetNode(node_->input(1)); in IsAlongAxis() local
1905 if (!IsConstant(*axis_node)) { in IsAlongAxis()
1908 if (HasAttribute(*axis_node, "value").ok()) { in IsAlongAxis()
1910 auto success = tensor.FromProto(axis_node->attr().at({"value"}).tensor()); in IsAlongAxis()
|