Searched refs:GetOutputSize (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | shape_test.cc | 45 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize() function in tflite::__anond412ad240111::ShapeOpModel
76 EXPECT_EQ(model.GetOutputSize(), 0); in TEST()
|
| D | batch_to_space_nd_test.cc | 47 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize() function in tflite::__anonbe27a5930111::BatchToSpaceNDOpModel
145 EXPECT_THAT(m.GetOutputSize(), 0); in TEST()
|
| D | one_hot_test.cc | 56 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize() function in tflite::__anon6d43443f0111::OneHotOpModel
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 161 int64 GetOutputSize(const int64 input, const int64 filter, const int64 stride, in GetOutputSize() function
738 int64 ox = GetOutputSize(ix, kx, sx, padding); in ConvolutionDimensionsFromInputs()
739 int64 oy = GetOutputSize(iy, ky, sy, padding); in ConvolutionDimensionsFromInputs()
1814 int64 ox = GetOutputSize(ix, kx, sx, padding); in OpDimensionsFromInputs()
1815 int64 oy = GetOutputSize(iy, ky, sy, padding); in OpDimensionsFromInputs()
|
| D | op_level_cost_estimator_test.cc | 322 int GetOutputSize(const int x, const int k, const int s, in GetOutputSize() function
356 int ho = GetOutputSize(h, kx, sx, padding); in GetPoolingOutputSize()
357 int wo = GetOutputSize(w, ky, sy, padding); in GetPoolingOutputSize()
|