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.h71 bool GetTensorFromConstNode(const string& node_name_or_input, Tensor* tensor);
Dconstant_folding.cc384 bool ConstantFolding::GetTensorFromConstNode(const string& node_name_or_input, in GetTensorFromConstNode() function in tensorflow::grappler::ConstantFolding
2155 if (GetTensorFromConstNode(node->input(1), &permutation_tensor) && in RemoveShuffleOrTranspose()
2208 GetTensorFromConstNode(node->input(1), &axis)) { in RemoveReverse()
2247 GetTensorFromConstNode(node->input(1), &begin) && in SimplifySlice()
2248 GetTensorFromConstNode(node->input(2), &size)) { in SimplifySlice()
2298 if (!GetTensorFromConstNode(node->input(i), &input_tensors[i - 1])) { in SimplifyStridedSlice()
2370 GetTensorFromConstNode(node->input(1), &multiplies)) { in SimplifyTile()
2395 if (GetTensorFromConstNode(node->input(1), &paddings)) { in SimplifyPad()
3796 if (!GetTensorFromConstNode(node.input(axis_idx), &axis_tensor)) { in GetConcatAxis()
Darithmetic_optimizer.cc340 bool GetTensorFromConstNode(const string& node_name_or_input, in GetTensorFromConstNode() function in tensorflow::grappler::__anon912b57ca0111::ArithmeticOptimizerStage
2800 if (!GetTensorFromConstNode(node->name(), &multiples)) { in CreateUpdatedMultiplesProto()
2827 if (!GetTensorFromConstNode(node->name(), &shape)) { in CreateUpdatedShapeProto()
3178 if (!GetTensorFromConstNode(node->input(1), &pow)) return Status::OK(); in TrySimplify()
3353 if (GetTensorFromConstNode(add_node->input(j), &constant)) { in TrySimplifyInternal()
3422 if (!GetTensorFromConstNode(node->input(1), &constant)) return Status::OK(); in TrySimplify()
4215 if (!GetTensorFromConstNode(node.input(axis_input), &axis_tensor)) in IsAxis0()