Searched refs:TfLiteTensorDim (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_test.c | 93 ASSERT_EQ(TfLiteTensorDim(input_tensor, 0), 2); in TestSmokeTest()
115 ASSERT_EQ(TfLiteTensorDim(output_tensor, 0), 2); in TestSmokeTest()
|
| D | c_api_test.cc | 63 EXPECT_EQ(TfLiteTensorDim(input_tensor, 0), 2); in TEST()
85 EXPECT_EQ(TfLiteTensorDim(output_tensor, 0), 2); in TEST()
125 EXPECT_EQ(TfLiteTensorDim(input_tensor, 0), 2); in TEST()
|
| D | c_api.h | 249 TFL_CAPI_EXPORT extern int32_t TfLiteTensorDim(const TfLiteTensor* tensor,
|
| D | c_api.cc | 192 int32_t TfLiteTensorDim(const TfLiteTensor* tensor, int32_t dim_index) { in TfLiteTensorDim() function
|
| /external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/ |
| D | Interpreter.cs | 150 dimensions[i] = TfLiteTensorDim(tensor, i); in GetTensorInfo()
263 private static extern int TfLiteTensorDim(TfLiteTensor tensor, int dim_index); in TfLiteTensorDim() method in TensorFlowLite.Interpreter
|
| /external/tensorflow/tensorflow/lite/swift/Sources/ |
| D | Interpreter.swift | 141 let dimensions = (0..<rank).map { Int(TfLiteTensorDim(cTensor, $0)) } in input()
187 let dimensions = (0..<rank).map { Int(TfLiteTensorDim(cTensor, $0)) } in output()
|
| /external/tensorflow/tensorflow/lite/objc/sources/ |
| D | TFLInterpreter.mm | 314 int32_t dimension = TfLiteTensorDim(cTensor, dimIndex);
|