Searched refs:GetOutputSize (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | shape_test.cc | 45 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize() function in tflite::__anonac5345a20111::ShapeOpModel
76 EXPECT_EQ(model.GetOutputSize(), 0); in TEST()
|
| D | one_hot_test.cc | 56 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize() function in tflite::__anon580bf43d0111::OneHotOpModel
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 139 int64 GetOutputSize(const int64 input, const int64 filter, const int64 stride, in GetOutputSize() function
622 int64 ox = GetOutputSize(ix, kx, sx, padding); in ConvolutionDimensionsFromInputs()
623 int64 oy = GetOutputSize(iy, ky, sy, padding); in ConvolutionDimensionsFromInputs()
1475 int64 ox = GetOutputSize(ix, kx, sx, padding); in OpDimensionsFromInputs()
1476 int64 oy = GetOutputSize(iy, ky, sy, padding); in OpDimensionsFromInputs()
|
| D | op_level_cost_estimator_test.cc | 289 int GetOutputSize(const int x, const int k, const int s, in GetOutputSize() function
323 int ho = GetOutputSize(h, kx, sx, padding); in GetPoolingOutputSize()
324 int wo = GetOutputSize(w, ky, sy, padding); in GetPoolingOutputSize()
|