Home
last modified time | relevance | path

Searched defs:TensorList (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_node.h88 struct TORCH_API TensorList : public TsNode { struct
93 TensorList() = delete; argument
96 bool CanBeReused(OpList values) const { in CanBeReused()
H A Dts_node.cpp82 TensorList::TensorList(OpList values) in TensorList() function in torch::lazy::TensorList
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_list.h67 TensorList() : tensors_(new Tensors) {} in TensorList() function
70 TensorList(const TensorList& other) in TensorList() function
78 TensorList(TensorList&& rhs) in TensorList() function
H A Dtensor_list_util.h25 class TensorList; variable
/aosp_15_r20/external/executorch/runtime/core/exec_aten/
H A Dexec_aten.h58 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 Dvariant_test.cc533 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 DTypeDefault.h19 using TensorList = ArrayRef<Tensor>; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DATen_fwd.h30 using TensorList = c10::ArrayRef<Tensor>; variable
/aosp_15_r20/external/executorch/exir/dialects/edge/arg/
H A Dtype.py13 TensorList = "Tensor[]" variable in ArgType
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/
H A Dtensor_list.py39 class TensorList(object): class
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_unbind_copy.cpp20 using TensorList = exec_aten::TensorList; typedef
H A Dop_split_copy.cpp20 using TensorList = exec_aten::TensorList; typedef
H A Dop_split_with_sizes_copy.cpp21 using TensorList = exec_aten::TensorList; typedef
/aosp_15_r20/external/executorch/devtools/etdump/
H A Dschema_flatcc.py30 class TensorList: class
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_expression.cc60 XlaExpression XlaExpression::TensorList(xla::XlaOp tensor_list) { in TensorList() function in tensorflow::XlaExpression
/aosp_15_r20/external/executorch/exir/
H A Dschema.py135 class TensorList: class
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DTensorBody.h70 using TensorList = ArrayRef<Tensor>; variable
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_test.cc453 TEST_F(TestKernelAttr, TensorList) { in TEST_F() argument
H A Dc_api_test.cc2394 TEST_F(CApiAttributesTest, TensorList) { in TEST_F() argument
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py428 class TensorList(torch.nn.Module): class
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py9298 TensorList = enum.auto() variable in ForeachRightmostArgType