Searched refs:kOutputShapeTensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | sparse_to_dense.cc | 32 constexpr int kOutputShapeTensor = 1; variable
151 context, GetInputSafe(context, node, kOutputShapeTensor, &output_shape)); in Prepare()
207 context, GetInputSafe(context, node, kOutputShapeTensor, &output_shape)); in SparseToDenseImpl()
|
| H A D | transpose_conv.cc | 48 constexpr int kOutputShapeTensor = 0; variable
253 context, GetInputSafe(context, node, kOutputShapeTensor, &output_shape)); in Prepare()
599 context, GetInputSafe(context, node, kOutputShapeTensor, &output_shape)); in Eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| H A D | convolution_op_builder.cc | 330 const int kOutputShapeTensor = 0; // Only used for TransposeConv in IsConvolutionOpSupported() local
345 if (!IsConstantTensor(GetInput(context, node, kOutputShapeTensor))) { in IsConvolutionOpSupported()
|