Searched defs:bias_t (Results 1 – 11 of 11) sorted by relevance
94 const at::Tensor& bias_t = bias.value().toTensor(); in fuseFrozenConvAddReluImpl() local
644 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest() local751 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest() local905 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest() local1301 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({128}); in RunTest() local1439 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({64}); in RunTest() local1540 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local1640 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({64}); in RunTest() local1932 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local2003 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local2096 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local[all …]
583 auto bias_t = GenerateRandomTensor<DTYPE>({128 * 3}); in RelpaceAddWithBiasAddDepthwiseConv2D() local689 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({64}); in RunTest() local
974 auto bias_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2})); in TEST_F() local
208 const std::optional<Tensor>& bias_t, in cudnn_convolution_relu()221 const std::optional<Tensor>& bias_t, in cudnn_convolution_add_relu()
708 const std::optional<Tensor>& bias_t, in cudnn_convolution_relu()762 const std::optional<Tensor>& bias_t, in cudnn_convolution_add_relu()
133 const Tensor& bias_t = *bias_t_maybe_owned; in cudnn_batch_norm() local
66 const Tensor& bias_t = *bias_t_maybe_owned; in miopen_batch_norm() local
804 const Tensor& bias_t = *bias_t_maybe_owned; in miopen_convolution() local905 const Tensor& bias_t = *bias_t_maybe_owned; in miopen_depthwise_convolution() local1472 const Tensor& bias_t = *bias_t_maybe_owned; in miopen_convolution_transpose() local
127 Tensor bias_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBias() local242 Tensor bias_t = MakeRandomTensor<T>({out_depth}); local
1295 const Tensor& bias_t, in convolution_jvp()1333 const Tensor& bias_t, in _convolution_jvp()6159 const Tensor& bias_t) { in _affine_jvp()6195 const Tensor& bias_t, in batch_norm_jvp()6250 const Tensor& bias_t, in layer_norm_jvp()6289 const Tensor& bias_t, in group_norm_jvp()