| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_list.h | 65 class TensorList {
67 TensorList() : tensors_(new Tensors) {} in TensorList() function
68 ~TensorList();
70 TensorList(const TensorList& other) in TensorList() function
78 TensorList(TensorList&& rhs) in TensorList() function
86 TensorList& operator=(const TensorList& rhs) {
97 TensorList& operator=(TensorList&& rhs) {
130 TensorList Copy() const { in Copy()
131 TensorList out; in Copy()
155 static_assert(Variant::CanInlineType<TensorList>() || sizeof(void*) < 8,
|
| D | list_kernels.cc | 66 const TensorList& tensor_list, int index, in GetElementShapeFromInput()
76 Status GetInputList(OpKernelContext* c, int index, const TensorList** list) { in GetInputList()
81 const TensorList* l = c->input(index).scalar<Variant>()().get<TensorList>(); in GetInputList()
93 const TensorList& input_list, in ForwardInputOrCreateNewList()
94 TensorList** output_list) { in ForwardInputOrCreateNewList()
103 TensorList* tmp_out = output_tensor->scalar<Variant>()().get<TensorList>(); in ForwardInputOrCreateNewList()
125 *output_list = output_tensor->scalar<Variant>()().get<TensorList>(); in ForwardInputOrCreateNewList()
146 TensorList empty; in Compute()
188 const TensorList* l = nullptr; in Compute()
210 TensorList* output_list = nullptr; in Compute()
[all …]
|
| D | tensor_list.cc | 24 TensorList::~TensorList() { in ~TensorList()
28 void TensorList::Encode(VariantTensorData* data) const { in Encode()
55 const TensorList& from, TensorList* to, in TensorListDeviceCopy()
71 INTERNAL_REGISTER_UNARY_VARIANT_DEVICE_COPY_FUNCTION(TensorList, DIRECTION, \
78 REGISTER_UNARY_VARIANT_DECODE_FUNCTION(TensorList, TensorList::kTypeName);
80 bool TensorList::Decode(const VariantTensorData& data) { in Decode()
125 const char TensorList::kTypeName[] = "tensorflow::TensorList";
|
| D | list_kernels.h | 48 const TensorList& tensor_list, int index,
51 Status GetInputList(OpKernelContext* c, int index, const TensorList** list);
55 const TensorList& input_list,
56 TensorList** output_list);
69 const TensorList* tensor_list = nullptr; in Compute()
169 const TensorList* l = nullptr; in Compute()
234 const TensorList* l = nullptr; in Compute()
268 TensorList* output_list = nullptr; in Compute()
297 const TensorList* tensor_list = nullptr; in Compute()
488 TensorList output_list; in Compute()
[all …]
|
| D | list_kernels.cu.cc | 122 TensorList,
125 DEVICE_GPU, TensorList,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TensorListScatter.pbtxt | 3 summary: "Creates a TensorList by indexing into a Tensor."
5 Each member of the TensorList corresponds to one row of the input tensor,
12 output_handle: The TensorList.
|
| D | api_def_TensorListScatterV2.pbtxt | 3 summary: "Creates a TensorList by indexing into a Tensor."
5 Each member of the TensorList corresponds to one row of the input tensor,
15 output_handle: The TensorList.
|
| D | api_def_TensorListGather.pbtxt | 3 summary: "Creates a Tensor by indexing into the TensorList."
5 Each row in the produced Tensor corresponds to the element in the TensorList
|
| D | api_def_TensorListScatterIntoExistingList.pbtxt | 5 Each member of the TensorList corresponds to one row of the input tensor,
11 output_handle: The TensorList.
|
| D | api_def_SendTPUEmbeddingGradients.pbtxt | 7 A TensorList of gradients with which to update embedding tables.
18 A TensorList of float32 scalars, one for each dynamic learning
|
| D | api_def_TensorListFromTensor.pbtxt | 3 summary: "Creates a TensorList which, when stacked, has the value of `tensor`."
|
| D | api_def_RecvTPUEmbeddingActivations.pbtxt | 7 A TensorList of embedding activations containing one Tensor per
|
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | tensor_list_test.py | 52 l = tl.TensorList(self._shape(()), dtypes.int32)
59 l = tl.TensorList(a.shape, a.dtype)
74 l = tl.TensorList(a.shape, a.dtype)
83 l = tl.TensorList(a.shape, a.dtype)
102 l = tl.TensorList(a.shape, a.dtype)
|
| D | tensor_list.py | 43 class TensorList(object): class
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | tensorlist_patterns.td | 21 // TensorList transformation patterns.
22 // Note that the pattern below rewrites `TensorList` tensors (which has type DT_VARIANT)
23 // into regular tensors. We also assume that each element in the `TensorList` has
|
| D | lower_static_tensor_list.cc | 192 const tensorflow::TensorList *list = in matchAndRewrite()
193 tensor.scalar<tensorflow::Variant>()().get<tensorflow::TensorList>(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_test.cc | 533 struct TensorList { struct
549 TensorList vec; in TEST()
566 const TensorList& stored_vec = *x.get<TensorList>(); in TEST()
577 Variant y = TensorList(); in TEST()
580 const TensorList& decoded_vec = *y.get<TensorList>(); in TEST()
|
| D | types.proto | 85 // "tensorflow::TensorList" in the variant type registry.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression.h | 72 static XlaExpression TensorList(xla::XlaOp tensor_list);
|
| D | xla_expression.cc | 59 XlaExpression XlaExpression::TensorList(xla::XlaOp tensor_list) { in TensorList() function in tensorflow::XlaExpression
|
| D | xla_op_kernel.cc | 600 SetOutputExpression(index, XlaExpression::TensorList(handle)); in SetTensorListOutput()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | known_issues.md | 9 `Support for TensorList crossing the XLA/TF boundary is not implemented`.
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 62 // Returns type of the TensorList element produced by this op.
66 // the TensorList element as a subtype.
929 inputs: A TensorList of gradients with which to update embedding tables.
935 learning_rates: A TensorList of float32 scalars, one for each dynamic learning
960 The gradients argument is a TensorList having the same length and shapes as the
966 gradients: A TensorList of gradients with which to update embedding tables.
967 learning_rates: A TensorList of learning rates used for updating the embedding
968 tables via the optimizer. The length of the TensorList must be equal to the
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | BUILD | 164 # TensorList ops are not implemented in the on-demand compilation model yet.
1351 # TensorList ops are not implemented in the on-demand compilation model yet.
|
| /external/llvm-project/mlir/docs/ |
| D | ShapeInference.md | 147 richer types (e.g., `TensorList<F32>`);
|