/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
H A D | ts_node.h | 88 struct TORCH_API TensorList : public TsNode { struct 93 TensorList() = delete; argument 96 bool CanBeReused(OpList values) const { in CanBeReused()
|
H A D | ts_node.cpp | 82 TensorList::TensorList(OpList values) in TensorList() function in torch::lazy::TensorList
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | tensor_list.h | 67 TensorList() : tensors_(new Tensors) {} in TensorList() function 70 TensorList(const TensorList& other) in TensorList() function 78 TensorList(TensorList&& rhs) in TensorList() function
|
H A D | tensor_list_util.h | 25 class TensorList; variable
|
/aosp_15_r20/external/executorch/runtime/core/exec_aten/ |
H A D | exec_aten.h | 58 using TensorList = at::TensorList; variable 111 using TensorList = ArrayRef<Tensor>; variable 152 using TensorList = exec_aten::TensorList; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | variant_test.cc | 533 struct TensorList { struct 534 void Encode(VariantTensorData* data) const { data->tensors_ = vec; } in Encode() 536 bool Decode(VariantTensorData data) { in Decode() 541 string TypeName() const { return "TensorList"; } in TypeName() 543 std::vector<Tensor> vec;
|
/aosp_15_r20/external/pytorch/aten/src/ATen/ |
H A D | TypeDefault.h | 19 using TensorList = ArrayRef<Tensor>; variable
|
/aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
H A D | ATen_fwd.h | 30 using TensorList = c10::ArrayRef<Tensor>; variable
|
/aosp_15_r20/external/executorch/exir/dialects/edge/arg/ |
H A D | type.py | 13 TensorList = "Tensor[]" variable in ArgType
|
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/ |
H A D | tensor_list.py | 39 class TensorList(object): class
|
/aosp_15_r20/external/executorch/kernels/portable/cpu/ |
H A D | op_unbind_copy.cpp | 20 using TensorList = exec_aten::TensorList; typedef
|
H A D | op_split_copy.cpp | 20 using TensorList = exec_aten::TensorList; typedef
|
H A D | op_split_with_sizes_copy.cpp | 21 using TensorList = exec_aten::TensorList; typedef
|
/aosp_15_r20/external/executorch/devtools/etdump/ |
H A D | schema_flatcc.py | 30 class TensorList: class
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_expression.cc | 60 XlaExpression XlaExpression::TensorList(xla::XlaOp tensor_list) { in TensorList() function in tensorflow::XlaExpression
|
/aosp_15_r20/external/executorch/exir/ |
H A D | schema.py | 135 class TensorList: class
|
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/ |
H A D | TensorBody.h | 70 using TensorList = ArrayRef<Tensor>; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/c/ |
H A D | kernels_test.cc | 453 TEST_F(TestKernelAttr, TensorList) { in TEST_F() argument
|
H A D | c_api_test.cc | 2394 TEST_F(CApiAttributesTest, TensorList) { in TEST_F() argument
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_modules.py | 428 class TensorList(torch.nn.Module): class
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_methods_invocations.py | 9298 TensorList = enum.auto() variable in ForeachRightmostArgType
|