Home
last modified time | relevance | path

Searched refs:GetTensorFromConstNode (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h74 bool GetTensorFromConstNode(const string& node_name_or_input, Tensor* tensor);
Dconstant_folding.cc380 bool ConstantFolding::GetTensorFromConstNode(const string& node_name_or_input, in GetTensorFromConstNode() function in tensorflow::grappler::ConstantFolding
2139 if (GetTensorFromConstNode(node->input(1), &permutation_tensor) && in RemoveShuffleOrTranspose()
2192 GetTensorFromConstNode(node->input(1), &axis)) { in RemoveReverse()
2231 GetTensorFromConstNode(node->input(1), &begin) && in SimplifySlice()
2232 GetTensorFromConstNode(node->input(2), &size)) { in SimplifySlice()
2282 if (!GetTensorFromConstNode(node->input(i), &input_tensors[i - 1])) { in SimplifyStridedSlice()
2354 GetTensorFromConstNode(node->input(1), &multiplies)) { in SimplifyTile()
2379 if (GetTensorFromConstNode(node->input(1), &paddings)) { in SimplifyPad()
3774 if (!GetTensorFromConstNode(node.input(axis_idx), &axis_tensor)) { in GetConcatAxis()
Darithmetic_optimizer.cc278 bool GetTensorFromConstNode(const string& node_name_or_input, in GetTensorFromConstNode() function in tensorflow::grappler::__anon73f394250111::ArithmeticOptimizerStage
2580 if (!GetTensorFromConstNode(node->input(1), &pow)) return Status::OK(); in TrySimplify()
2755 if (GetTensorFromConstNode(add_node->input(j), &constant)) { in TrySimplifyInternal()
2824 if (!GetTensorFromConstNode(node->input(1), &constant)) return Status::OK(); in TrySimplify()
3590 if (!GetTensorFromConstNode(node.input(axis_input), &axis_tensor)) in IsAxis0()