Home
last modified time | relevance | path

Searched refs:empty_tensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Dpatcher.py88 empty_tensor = f.get_slice(k)
90 tuple(empty_tensor.get_shape()),
92 empty_tensor.get_dtype()
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_common_rules.py27 empty_tensor = torch.empty(shape)
29 empty_tensor.shape,
30 empty_tensor.stride(),
31 empty_tensor.dtype,
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_io.cpp44 Eigen::Tensor<double,1,DataLayout> empty_tensor(0); in test_output_1d() local
46 empty_os << empty_tensor; in test_output_1d()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dregister_packed_params.cpp126 auto empty_tensor = in RegisterPrePackParams() local
129 none_node->t_(attr::value, empty_tensor); in RegisterPrePackParams()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dworker.cc128 static Tensor empty_tensor(DT_FLOAT); in PrepareRunGraph() local
137 out->insert({req->recv_key(i), empty_tensor}); in PrepareRunGraph()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp1017 static tensor empty_tensor; in linear_int8_with_onednn_weight() local
1021 empty_tensor; in linear_int8_with_onednn_weight()
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py4301 empty_tensor = torch.tensor([])
4304 self.run_test(model, (empty_tensor, empty_idx, empty_tensor))
11498 empty_tensor = torch.tensor([], dtype=torch.float).view(0, 0, 0, 0, 0)
11501 (random_data, empty_tensor),
11505 self.run_test(model, (random_data, empty_tensor), remained_onnx_input_idx=[])
11555 empty_tensor = torch.tensor([], dtype=torch.float).view(0, 0, 0, 0, 0)
11559 (random_data, empty_tensor),
11566 (random_data, empty_tensor),
11572 self.run_test(model, (random_data, empty_tensor), remained_onnx_input_idx=[])
11611 empty_tensor = torch.tensor([], dtype=torch.float).view(0, 0, 0, 0, 0)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc854 typename TTypes<T, 4>::ConstTensor empty_tensor(nullptr, 0, 0, 0, 0); in operator ()() local
858 empty_tensor, epsilon, activation_mode, in operator ()()