Searched defs:filter_t (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_conv_ops_test.cc | 106 Tensor filter_t = GetRandomFilterTensor(dims); in DefaultConv2D() local 127 Tensor filter_t = GetRandomFilterTensor(dims); in MklConv2D() local 154 Tensor filter_t = GetRandomFilterTensor(dims); in DefaultConv2DBwdInput() local 181 Tensor filter_t = GetRandomFilterTensor(dims); in MklConv2DBwdInput() local 216 Tensor filter_t = GetRandomFilterTensor(dims); in DefaultConv2DBwdFilter() local 244 Tensor filter_t = GetRandomFilterTensor(dims); in MklConv2DBwdFilter() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | remapper_test.cc | 643 auto filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 3, 128}); in RunTest() local 750 auto filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 3, 128}); in RunTest() local 904 auto filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 3, 128}); in RunTest() local 1037 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 1, 3, 6}); in TEST_F() local 1103 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 1, 3, 6}); in TEST_F() local 1169 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 6}); in TEST_F() local 1300 auto filter_t = GenerateTensorWithSetRandom<DTYPE>({1, 1, 3, 128}); in RunTest() local 1539 auto filter_t = GenerateRandomTensor<DT_FLOAT>({3, 3, 3, 128}); in TEST_F() local 1749 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local 1848 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local [all …]
|
H A D | mkl_remapper_test.cc | 582 auto filter_t = GenerateRandomTensor<DTYPE>({1, 1, 3, 128}); in RelpaceAddWithBiasAddDepthwiseConv2D() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | conv_ops_benchmark_test.cc | 94 Tensor filter_t = in Conv2D() local 240 Tensor filter_t = local 283 Tensor filter_t = local
|
H A D | conv_grad_input_ops.h | 180 auto filter_t = filter.tensor<T, 4>(); in operator() local
|
H A D | conv_grad_filter_ops_benchmark_test.cc | 49 Tensor filter_t = in Conv2DBackpropFilter() local
|
H A D | conv_grad_input_ops_benchmark_test.cc | 48 Tensor filter_t = in Conv2DBackpropInput() local
|