Searched refs:ExtractShape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | shape.cc | 30 void ExtractShape(const TfLiteTensor* input, OutType* output_data) { in ExtractShape() function
72 ExtractShape(input, GetTensorData<int32_t>(output)); in Eval()
75 ExtractShape(input, GetTensorData<int64_t>(output)); in Eval()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 508 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape() function
566 if (!ExtractShape(*shape_node1, properties, &shape1, &min_id)) { in MaterializeBroadcastGradientArgs()
570 if (!ExtractShape(*shape_node2, properties, &shape2, &min_id)) { in MaterializeBroadcastGradientArgs()
|