Searched refs:kShapeTensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape.cc | 31 constexpr int kShapeTensor = 1; variable
75 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in GetOutputShapeFromTensor()
109 const TfLiteTensor* shape = GetInput(context, node, kShapeTensor); in ShapeIsVector()
135 IsConstantTensor(GetInput(context, node, kShapeTensor))) { in Prepare()
|
| D | broadcast_to.cc | 32 constexpr int kShapeTensor = 1; variable
39 shape = GetInput(context, node, kShapeTensor); in BroadcastToContext()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | reshape_op_builder.cc | 128 const int kShapeTensor = 1; in IsReshapeOpSupported() local
130 TF_LITE_ENSURE_OK(context, GetInputSafe(context, node, kShapeTensor, &shape)); in IsReshapeOpSupported()
|