Home
last modified time | relevance | path

Searched refs:out_tensor (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dsummary_op_test.cc68 Tensor* out_tensor = GetOutput(0); in TEST_F() local
69 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
71 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
88 Tensor* out_tensor = GetOutput(0); in TEST_F() local
89 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
91 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
108 Tensor* out_tensor = GetOutput(0); in TEST_F() local
109 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
111 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
177 Tensor* out_tensor = GetOutput(0); in TEST_F() local
[all …]
Dsummary_image_op_test.cc87 Tensor* out_tensor = GetOutput(0); in TEST_F() local
88 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
90 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
110 Tensor* out_tensor = GetOutput(0); in TEST_F() local
111 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
113 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
142 Tensor* out_tensor = GetOutput(0); in TEST_F() local
143 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
145 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
Dsummary_audio_op_test.cc93 Tensor* out_tensor = GetOutput(0); in TEST_F() local
94 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
96 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
127 Tensor* out_tensor = GetOutput(0); in TEST_F() local
128 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
130 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
Dcast_op.h34 typename TTypes<OUT_TYPE>::Flat out_tensor, \
38 out_tensor.device(d) = \
42 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
52 typename TTypes<OUT_TYPE>::Flat out_tensor, \
56 out_tensor.device(d) = \
60 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
79 typename TTypes<OUT_TYPE>::Flat out_tensor, \
82 out_tensor.device(d) = in_tensor.template cast<OUT_TYPE>(); \
Dsummary_tensor_op_test.cc80 Tensor* out_tensor = GetOutput(0); in TEST_F() local
81 ASSERT_EQ(0, out_tensor->dims()); in TEST_F()
83 ParseProtoUnlimited(&summary, out_tensor->scalar<string>()()); in TEST_F()
Dcount_up_to_op.cc48 Tensor* out_tensor; in Compute() local
50 &out_tensor)); in Compute()
51 out_tensor->scalar<T>()() = before_increment; in Compute()
Dhistogram_op.cc106 Tensor* out_tensor; in Compute() local
108 ctx->allocate_output(0, TensorShape({nbins}), &out_tensor)); in Compute()
109 auto out = out_tensor->flat<Tout>(); in Compute()
Dsparse_tensor_dense_add_op.cc93 auto out_tensor = out_t->tensor<T, N>(); \ in Compute()
94 out_tensor.device(ctx->eigen_device<Device>()) = b->tensor<T, N>(); \ in Compute()
99 out_tensor); \ in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc27 Tensor* out_tensor) { in ParseTensorProtoToTensor() argument
31 *out_tensor = parsed; in ParseTensorProtoToTensor()
61 Status InMemoryRunStepRequest::FeedValue(size_t i, Tensor* out_tensor) const { in FeedValue()
62 *out_tensor = feeds_[i].second; in FeedValue()
67 TensorProto* out_tensor) const { in FeedValue()
68 feeds_[i].second.AsProtoTensorContent(out_tensor); in FeedValue()
155 Tensor* out_tensor) const { in FeedValue()
156 if (!ParseTensorProtoToTensor(request_.feed(i).tensor(), out_tensor)) { in FeedValue()
164 TensorProto* out_tensor) const { in FeedValue()
165 *out_tensor = request_.feed(i).tensor(); in FeedValue()
[all …]
Dmessage_wrappers.h66 virtual Status FeedValue(size_t i, Tensor* out_tensor) const = 0;
67 virtual Status FeedValue(size_t i, TensorProto* out_tensor) const = 0;
122 Status FeedValue(size_t i, Tensor* out_tensor) const override;
123 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
173 Status FeedValue(size_t i, Tensor* out_tensor) const override;
174 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
214 Status FeedValue(size_t i, Tensor* out_tensor) const override;
215 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
276 virtual Status SendValue(size_t i, Tensor* out_tensor) const = 0;
335 Status SendValue(size_t i, Tensor* out_tensor) const override;
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc117 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
118 ASSERT_EQ(out_tensor.dtype(), tensorflow::DT_FLOAT); in TEST()
119 ASSERT_EQ(out_tensor.NumElements(), 6); in TEST()
120 ASSERT_THAT(GetTensorShape(out_tensor), ElementsAre(1, 2, 1, 3)); in TEST()
135 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
136 ASSERT_EQ(out_tensor.dtype(), tensorflow::DT_STRING); in TEST()
137 ASSERT_EQ(out_tensor.NumElements(), 6); in TEST()
138 ASSERT_THAT(GetTensorShape(out_tensor), ElementsAre(1, 2, 1, 3)); in TEST()
180 tensorflow::Tensor out_tensor = buffer_map.GetTensor(0); in TEST() local
181 ASSERT_EQ(out_tensor.dtype(), tensorflow::DT_FLOAT); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/python/
Dlite_test.py90 out_tensor = in_tensor + in_tensor
95 [out_tensor])
119 out_tensor = array_ops.reshape(in_tensor, shape=[2, 2])
124 [out_tensor])
151 out_tensor = array_ops.fake_quant_with_min_max_args(
157 sess, [in_tensor_1, in_tensor_2], [out_tensor])
196 out_tensor = array_ops.fake_quant_with_min_max_args(
202 sess, [in_tensor_1, in_tensor_2], [out_tensor])
216 out_tensor = in_tensor_final + in_tensor_final
221 [out_tensor])
[all …]
Dconvert_saved_model_test.py125 out_tensor = in_tensor + in_tensor
127 outputs = {"y": out_tensor}
139 out_tensor = in_tensor_1 + in_tensor_2
141 outputs = {"z": out_tensor}
297 out_tensor = in_tensor + in_tensor
300 sig_output_tensor = saved_model.utils.build_tensor_info(out_tensor)
Dlite_flex_test.py36 out_tensor = in_tensor + in_tensor
41 [out_tensor])
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py577 for out_tensor in op.outputs:
578 consumers += [consumer_op for consumer_op in out_tensor.consumers()]
814 for out_tensor in op.outputs:
815 if out_tensor.name not in self._tensorname_idx_map:
817 'out_tensor %s is not in tensorname_idx_map'%out_tensor.name)
818 line += ' %d'%self._tensorname_idx_map[out_tensor.name]
1037 def _skip_tensor(self, op_id, out_tensor, user_included, argument
1047 if out_tensor.dtype in non_numeric_tensor_types:
1048 self._instrument_records[out_tensor.name] = TensorTracer.reason(
1052 if [consumer for consumer in out_tensor.consumers() if
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dmorphological_ops_test.py49 out_tensor = nn_ops.dilation2d(
56 self.assertAllClose(out, self.evaluate(out_tensor))
213 out_tensor = nn_ops.dilation2d(
220 out_shape = self.evaluate(out_tensor).shape
225 out_tensor,
325 out_tensor = nn_ops.erosion2d(
332 self.assertAllClose(out, self.evaluate(out_tensor))
489 out_tensor = nn_ops.erosion2d(
496 out_shape = self.evaluate(out_tensor).shape
501 out_tensor,
Dlistdiff_op_test.py48 out_tensor, idx_tensor = diff_func(x_tensor, y_tensor,
50 tf_out, tf_idx = self.evaluate([out_tensor, idx_tensor])
53 self.assertEqual(1, out_tensor.get_shape().ndims)
Dextract_volume_patches_op_test.py48 out_tensor = array_ops.extract_volume_patches(
54 self.assertAllClose(patches, self.evaluate(out_tensor))
Dextract_image_patches_op_test.py46 out_tensor = array_ops.extract_image_patches(
53 self.assertAllClose(patches, self.evaluate(out_tensor))
/external/tensorflow/tensorflow/c/
Dcheckpoint_reader.cc90 const string& name, std::unique_ptr<tensorflow::Tensor>* out_tensor, in GetTensor() argument
94 status = reader_->GetTensor(name, out_tensor); in GetTensor()
100 out_tensor->reset(new Tensor(dtype, shape)); in GetTensor()
101 status = v2_reader_->Lookup(name, out_tensor->get()); in GetTensor()
102 if (!status.ok()) out_tensor->reset(); in GetTensor()
/external/tensorflow/tensorflow/contrib/framework/kernels/
Dzero_initializer_op.cc50 Tensor* out_tensor = nullptr; in Compute() local
53 &out_persistent, &out_tensor, attr)); in Compute()
55 out_tensor->flat<T>()); in Compute()
56 ctx->replace_ref_input(0, *out_tensor, true); in Compute()
/external/tensorflow/tensorflow/compiler/tests/
Dlistdiff_op_test.py40 out_tensor, idx_tensor = array_ops.listdiff(
42 tf_out, tf_idx = self.evaluate([out_tensor, idx_tensor])
45 self.assertEqual(1, out_tensor.get_shape().ndims)
Dextract_image_patches_op_test.py50 out_tensor = array_ops.extract_image_patches(
58 self.assertAllClose(patches, out_tensor.eval(feed_dict=feed_dict))
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.h146 Tensor out_tensor; in OptionalBinaryAdd() local
148 BinaryAddTensors<Device>(ctx, a_tensor, b_tensor, &out_tensor)); in OptionalBinaryAdd()
149 out_tensors.push_back(std::move(out_tensor)); in OptionalBinaryAdd()
/external/tensorflow/tensorflow/contrib/training/python/training/
Dsampling_ops_threading_test.py59 out_tensor = queue.dequeue()
70 sess.run([out_tensor])

123