Searched refs:ExtractShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | shape.cc | 33 void ExtractShape(const TfLiteTensor* input, OutType* output_data) { in ExtractShape() function
81 ExtractShape(input, GetTensorData<int32_t>(output)); in Prepare()
84 ExtractShape(input, GetTensorData<int64_t>(output)); in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | shape.cc | 31 void ExtractShape(const TfLiteEvalTensor* input, int32_t* output_data) { in ExtractShape() function
54 ExtractShape(input, tflite::micro::GetTensorData<int32_t>(output)); in Eval()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 562 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape() function
620 if (!ExtractShape(*shape_node1, properties, &shape1, &min_id)) { in MaterializeBroadcastGradientArgs()
624 if (!ExtractShape(*shape_node2, properties, &shape2, &min_id)) { in MaterializeBroadcastGradientArgs()
|