Home
last modified time | relevance | path

Searched defs:z_t (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_conv_add_relu_fusion_cuda.cpp106 const at::Tensor& z_t = z.value().toTensor(); in fuseFrozenConvAddReluImpl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConvPlaceholders.cpp219 const Tensor& z_t, in cudnn_convolution_add_relu()
H A DConvShared.cpp760 const Tensor& z_t, in cudnn_convolution_add_relu()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc2438 auto z_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2, 2})); in TEST_F() local
2647 auto z_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({32, 32, 32})); in TEST_F() local