Home
last modified time | relevance | path

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

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