| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_old_batch_norms.cc | 236 NodeDef axis_node = concat_node_match.inputs[2].node; local
237 CHECK_EQ("Const", axis_node.op());
238 Tensor axis = GetNodeTensorAttr(axis_node, "value");
|
| sparsify_gather_test.cc | 62 NodeDef* axis_node = local
66 SetNodeTensorAttr<int32>("value", axis_t, axis_node);
67 CreateNode(name, "GatherV2", {params, indices, axis_node}, graph_def);
|
| sparsify_gather.cc | 301 const NodeDef& axis_node = match.inputs[2].node;
304 TF_RETURN_IF_ERROR(GetNodeAttr(axis_node, "value", &axis_t));
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 2298 NodeDef* axis_node = optimized_graph->add_node(); local
[all...] |
| layout_optimizer.cc | 1904 auto axis_node = node_map_->GetNode(node_->input(1)); local
[all...] |
