Home
last modified time | relevance | path

Searched refs:WrappedTensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_supported_values_test.py46 class WrappedTensor(composite_tensor.CompositeTensor): class
66 return WrappedTensor(self.value.__getitem__(idx))
84 return WrappedTensor
90 return WrappedTensor(value)
121 return WrappedTensor(value)
124 if isinstance(value, WrappedTensor):
132 if isinstance(value, WrappedTensor):
135 if any(isinstance(x, WrappedTensor) for x in value):
141 ragged_tensor._add_supported_value_type(WrappedTensor)
157 values = WrappedTensor(tensor_values)
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dvectorizer.h31 using VectorizerOutput = std::vector<WrappedTensor>;
37 VectorizerInput(std::vector<WrappedTensor>&& inputs) in VectorizerInput()
73 const WrappedTensor& at(int index) const { in at()
80 std::vector<WrappedTensor>::const_iterator begin() const { in begin()
84 std::vector<WrappedTensor>::const_iterator end() const { in end()
92 std::vector<WrappedTensor> inputs_;
Dwrapped_tensor.h25 struct WrappedTensor { struct
37 WrappedTensor(Node* node, int output_index, bool stacked) in WrappedTensor() argument
Dcwise_op_vectorizer.cc55 auto get_stacked_rank = [&scope](const WrappedTensor& input) { in ExpandDimsForBroadcast()
73 [&scope, &max_rank](const WrappedTensor& tensor, const Output& rank) { in ExpandDimsForBroadcast()
75 return WrappedTensor(tensor.node, tensor.output_index, false); in ExpandDimsForBroadcast()
105 return WrappedTensor(reshaped.node(), 0, true); in ExpandDimsForBroadcast()
Dvectorizer_registry_test.cc47 std::vector<WrappedTensor> inputs, outputs; in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc183 Status StackTensor(WrappedTensor* unstacked, TensorDesc* result);
208 std::map<TensorDesc, WrappedTensor> conversion_map_;
245 std::vector<WrappedTensor> inputs, outputs; in AddConversionMapping()
409 Status Vectorization::StackTensor(WrappedTensor* unstacked, in StackTensor()
590 conversion_map_.insert({{tensor.first, i}, WrappedTensor(node, i, false)}); in AddUnstackedTensorMappingsHelper()
593 WrappedTensor(node, Graph::kControlSlot, false)}); in AddUnstackedTensorMappingsHelper()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1033 WrappedTensor = collections.namedtuple("WrappedTensor", variable
1064 return WrappedTensor(tensor, is_stacked, is_sparse_stacked)
1353 assert all(isinstance(o, WrappedTensor) for o in outputs)
1411 if isinstance(output, WrappedTensor):
1427 assert isinstance(new_output, (WrappedTensor, ops.Operation)), new_output
1463 (WrappedTensor, ops.Operation))
1628 if isinstance(new_outputs, WrappedTensor):
1642 assert isinstance(new_output, WrappedTensor), (new_output, y, y_op)
4156 assert all(isinstance(arg, WrappedTensor) for arg in args), args