Home
last modified time | relevance | path

Searched defs:tensor (Results 1 – 25 of 298) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py100 def make_generator(tensor): argument
163 def fn(dataset, tensor, shape): argument
171 def fn(dataset, tensor, shape): argument
179 def fn(dataset, tensor, shape): argument
187 def fn(dataset, tensor, shape): argument
195 def fn(dataset, tensor, shape): argument
203 def fn(dataset, tensor, shape): argument
211 def fn(dataset, tensor, shape): argument
219 def fn(dataset, tensor, shape): argument
227 def fn(dataset, tensor, shape): argument
[all …]
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py105 def _subscribe_new(tensor, side_effects, control_cache): argument
156 def _subscribe_extend(tensor, side_effects): argument
183 def _is_subscribed_identity(tensor): argument
216 def _subscribe(tensor, side_effects, control_cache): argument
258 def _preserve_control_flow_context(tensor): argument
288 def _scoped_subscribe(tensor, side_effects, control_cache): argument
/external/eigen/unsupported/test/
Dcxx11_tensor_custom_index.cpp24 Tensor<float, 4, DataLayout> tensor(2, 3, 5, 7); in test_map_as_index() local
46 Tensor<float, 4, DataLayout> tensor(2, 3, 5, 7); in test_matrix_as_index() local
64 Tensor<float, 4, DataLayout> tensor(2, 3, 5, 7); in test_varlist_as_index() local
79 Tensor<float, 4, DataLayout> tensor(2, 3, 5, 7); in test_sizes_as_index() local
Dcxx11_tensor_argmax.cpp22 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_simple_index_tuples() local
39 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_index_tuples_dim() local
57 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_argmax_tuple_reducer() local
89 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_argmin_tuple_reducer() local
121 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_simple_argmax() local
142 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_simple_argmin() local
163 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_argmax_dim() local
221 Tensor<float, 4, DataLayout> tensor(2,3,5,7); in test_argmin_dim() local
Dcxx11_tensor_io.cpp19 Tensor<int, 0, DataLayout> tensor; in test_output_0d() local
33 Tensor<int, 1, DataLayout> tensor(5); in test_output_1d() local
55 Tensor<int, 2, DataLayout> tensor(5, 3); in test_output_2d() local
91 Tensor<std::string, 2, DataLayout> tensor(5, 3); in test_output_string() local
107 Tensor<int, 1, DataLayout> tensor(5); in test_output_const() local
Dcxx11_tensor_ifft.cpp19 Tensor<double, 1, DataLayout> tensor(sequence_length); in test_1D_fft_ifft_invariant() local
41 Tensor<double, 2, DataLayout> tensor(dim0, dim1); in test_2D_fft_ifft_invariant() local
69 Tensor<double, 3, DataLayout> tensor(dim0, dim1, dim2); in test_3D_fft_ifft_invariant() local
101 Tensor<double, 4, DataLayout> tensor(dim0, dim1, dim2, dim3); in test_sub_fft_ifft_invariant() local
Dcxx11_tensor_layout_swap.cpp18 Tensor<float, 3, ColMajor> tensor(2,3,7); in test_simple_swap() local
38 Tensor<float, 3, ColMajor> tensor(2,3,7); in test_swap_as_lvalue() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtensor.h28 inline float* GetTensorData(TfLiteTensor* tensor) { in GetTensorData()
33 inline uint8_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData()
38 inline int32_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData()
43 inline int64_t* GetTensorData(TfLiteTensor* tensor) { in GetTensorData()
75 inline Dims<4> GetTensorDims(const TfLiteTensor* tensor) { in GetTensorDims()
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Dsummaries.py42 def _get_summary_name(tensor, name=None, prefix=None, postfix=None): argument
63 def add_histogram_summary(tensor, name=None, prefix=None): argument
79 def add_image_summary(tensor, name=None, prefix=None, print_summary=False): argument
105 def add_scalar_summary(tensor, name=None, prefix=None, print_summary=False): argument
132 def add_zero_fraction_summary(tensor, name=None, prefix=None, argument
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Dconditioning_utils_impl.py40 def _get_shape(tensor): argument
47 def condition_tensor(tensor, conditioning): argument
91 def condition_tensor_from_onehot(tensor, one_hot_labels, embedding_size=256): argument
/external/tensorflow/tensorflow/contrib/lite/java/src/test/java/org/tensorflow/lite/
DTensorTest.java56 Tensor tensor = Tensor.fromHandle(nativeHandle); in testFromHandle() local
65 Tensor tensor = Tensor.fromHandle(nativeHandle); in testCopyTo() local
75 Tensor tensor = Tensor.fromHandle(nativeHandle); in testCopyToWrongType() local
92 Tensor tensor = Tensor.fromHandle(nativeHandle); in testCopyToWrongShape() local
/external/tensorflow/tensorflow/core/framework/
Dtensor_util.cc24 namespace tensor { namespace
53 for (const Tensor& tensor : tensors) { in Concat() local
79 for (const Tensor& tensor : tensors) { in Concat() local
95 for (const Tensor& tensor : tensors) { in Concat() local
109 Status Split(const Tensor& tensor, const gtl::ArraySlice<int64>& sizes, in Split()
Dtensor_testutil.h57 void FillValues(Tensor* tensor, gtl::ArraySlice<T> vals) { in FillValues()
67 void FillValues(Tensor* tensor, std::initializer_list<SrcType> vals) { in FillValues()
82 void FillIota(Tensor* tensor, const T& val) { in FillIota()
91 void FillFn(Tensor* tensor, std::function<T(int)> fn) { in FillFn()
Dunique_tensor_references.cc26 for (auto& tensor : refs) { in ~UniqueTensorReferences() local
33 void UniqueTensorReferences::Add(const Tensor& tensor) { in Add()
/external/tensorflow/tensorflow/contrib/lite/testing/
Dtf_driver.cc37 void FillTensorWithData(tensorflow::Tensor* tensor, const string& csv_values) { in FillTensorWithData()
47 void FillTensorWithZeros(tensorflow::Tensor* tensor) { in FillTensorWithZeros()
55 string TensorDataToCsvString(const tensorflow::Tensor& tensor) { in TensorDataToCsvString()
114 auto tensor = CreateTensor(input_types_[id], input_shapes_[id]); in SetInput() local
134 auto tensor = input_tensors_[input_names_[id]]; in ResetTensor() local
Dtflite_driver.cc68 bool Check(bool verbose, const TfLiteTensor& tensor) { in Check()
86 bool TypedCheck(bool verbose, const TfLiteTensor& tensor) { in TypedCheck()
154 auto* tensor = interpreter_->tensor(id); in ResetTensor() local
170 auto* tensor = interpreter_->tensor(id); in SetInput() local
205 auto* tensor = interpreter_->tensor(id); in SetExpectation() local
243 auto* tensor = interpreter_->tensor(id); in CheckResults() local
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dsummaries.py43 def _add_scalar_summary(tensor, tag=None): argument
61 def _add_histogram_summary(tensor, tag=None): argument
107 def summarize_tensor(tensor, tag=None): argument
/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator.h67 Status ValidateShape(const Tensor* tensor) in ValidateShape()
117 const Tensor** tensor) override in GetAndValidateTensorInputForApplyGrad()
127 void CleanUpGradTensor(const Tensor* tensor) override { in CleanUpGradTensor()
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops.py354 def _transpose_if_necessary(tensor, perm): argument
362 def _reshape_if_necessary(tensor, new_shape): argument
374 def _get_shape(tensor): argument
/external/tensorflow/tensorflow/contrib/quantize/python/
Dgraph_matcher.py28 def match(self, op, tensor): argument
66 def match(self, op, tensor): argument
95 def match(self, op, tensor): argument
132 def add(self, pattern, op, tensor): argument
189 def _match_pattern(self, pattern, op, tensor): argument
/external/tensorflow/tensorflow/contrib/lite/
Dstring_util.cc67 void DynamicBuffer::WriteToTensor(TfLiteTensor* tensor) { in WriteToTensor()
103 int GetStringCount(const TfLiteTensor* tensor) { in GetStringCount()
108 StringRef GetString(const TfLiteTensor* tensor, int string_index) { in GetString()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dkernel_util.h56 inline bool IsConstantTensor(TfLiteTensor* tensor) { in IsConstantTensor()
62 inline bool IsDynamicTensor(TfLiteTensor* tensor) { in IsDynamicTensor()
67 inline void SetTensorToDynamic(TfLiteTensor* tensor) { in SetTensorToDynamic()
/external/tensorflow/tensorflow/contrib/mpi_collectives/python/ops/
Dmpi_ops.py91 def _allreduce(tensor, name=None): argument
108 def allgather(tensor, name=None): argument
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dtensor_util.py194 def with_same_shape(expected_tensor, tensor): argument
211 def with_shape(expected_shape, tensor): argument
281 def assert_scalar_int(tensor, name=None): argument
/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_io_utils_test.cc88 Tensor tensor(DT_FLOAT, TensorShape({1, 1})); in TEST_F() local
110 Tensor tensor(DT_FLOAT, TensorShape({1, 1})); in TEST_F() local
130 Tensor tensor(DT_FLOAT, TensorShape({1, 1})); in TEST_F() local
149 Tensor tensor(DT_STRING, TensorShape({1, 1})); in TEST_F() local
164 Tensor tensor(DT_STRING, TensorShape({1, 2})); in TEST_F() local
185 Tensor tensor(DT_INT32, TensorShape({1, 1})); in TEST_F() local
265 Tensor tensor(DT_INT32, TensorShape({1, 1})); in TEST_F() local
313 Tensor tensor(DT_INT32, TensorShape({1, 1})); in TEST_F() local

12345678910>>...12