Home
last modified time | relevance | path

Searched defs:scale_t (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc162 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F() local
270 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F() local
399 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F() local
536 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F() local
1750 auto scale_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local
1849 auto scale_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local
H A Dmkl_remapper_test.cc354 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_benchmark_test.cc176 Tensor scale_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBatchNorm() local
285 Tensor scale_t = MakeRandomTensor<T>({out_depth}); local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h69 using scale_t = std::optional<double>; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp15 using scale_t = std::vector<std::optional<double>>; typedef
H A DUpSampleKernel.cpp24 using scale_t = std::vector<std::optional<double>>; typedef