| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding_test.cc | 48 Tensor result_tensor; in Validate() local
49 EXPECT_TRUE(result_tensor.FromProto(response.tensor())); in Validate()
50 EXPECT_EQ(t.dtype(), result_tensor.dtype()); in Validate()
51 EXPECT_EQ(t.shape().DebugString(), result_tensor.shape().DebugString()); in Validate()
52 EXPECT_EQ(t.DebugString(), result_tensor.DebugString()); in Validate()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | custom_gradient_test.cc | 120 TF_Tensor* result_tensor; in TEST_P() local
121 s = GetValue(outputs[0], &result_tensor); in TEST_P()
123 auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
126 TF_DeleteTensor(result_tensor); in TEST_P()
127 result_tensor = nullptr; in TEST_P()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients_test.cc | 145 TF_Tensor* result_tensor; in TEST_P() local
146 s = GetValue(outputs[1], &result_tensor); in TEST_P()
148 auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
151 TF_DeleteTensor(result_tensor); in TEST_P()
152 result_tensor = nullptr; in TEST_P()
|
| D | c_api_unified_experimental_test.cc | 96 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
97 float* result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
100 TF_DeleteTensor(result_tensor); in TEST_P()
161 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
165 memcpy(&result_data[0], TF_TensorData(result_tensor), in TEST_P()
166 TF_TensorByteSize(result_tensor)); in TEST_P()
173 TF_DeleteTensor(result_tensor); in TEST_P()
244 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
248 memcpy(&result_data[0], TF_TensorData(result_tensor), in TEST_P()
249 TF_TensorByteSize(result_tensor)); in TEST_P()
[all …]
|
| D | unified_api_testutil.h | 67 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor);
|
| D | unified_api_testutil.cc | 176 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor) { in GetValue() argument
182 *result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in GetValue()
|
| D | gradients_util.h | 52 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor);
|
| D | gradients_util.cc | 130 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor) { in GetValue() argument
136 *result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in GetValue()
|
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | misc_test.py | 68 for (i, j, k), result_tensor in results:
70 len(list(range(i, j, k))), self.evaluate(result_tensor))
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 293 auto* result_tensor = interpreter_->tensor(kResultTensorId); in GetLookupResult() local
294 auto result_tensor_shape = GetTensorShape(result_tensor); in GetLookupResult()
295 auto* result_tensor_data = GetTensorData<ValueType>(result_tensor); in GetLookupResult()
306 auto* result_tensor = interpreter_->tensor(kResultTwoTensorId); in GetLookupTwoResult() local
307 auto result_tensor_shape = GetTensorShape(result_tensor); in GetLookupTwoResult()
308 auto* result_tensor_data = GetTensorData<ValueType>(result_tensor); in GetLookupTwoResult()
319 auto* result_tensor = interpreter_->tensor(kResultTensorId); in GetStringLookupResult() local
320 auto result_tensor_shape = GetTensorShape(result_tensor); in GetStringLookupResult()
325 auto string_ref = GetString(result_tensor, i); in GetStringLookupResult()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_op.h | 139 Tensor result_tensor = ConvertDescriptorToTensor<OutputDataType>( in Compute() local
142 ctx->set_output(0, result_tensor); in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 792 def _check(self, result_tensor, result_np, input_sp_t): argument
793 self.assertTrue(isinstance(result_tensor, sparse_tensor.SparseTensor))
796 result_tensor.indices)
798 result_tensor.dense_shape)
801 result_tensor.indices, result_tensor.dense_shape, result_tensor.values)
|
| D | cwise_ops_unary_test.py | 121 def _check(self, result_tensor, result_np, input_sp_t, tol): argument
122 self.assertTrue(isinstance(result_tensor, sparse_tensor.SparseTensor))
124 self.assertAllEqual(input_sp_t.indices, result_tensor.indices)
125 self.assertAllEqual(input_sp_t.dense_shape, result_tensor.dense_shape)
127 self.assertAllClose(result_np, result_tensor.values)
129 self.assertAllClose(result_np, result_tensor.values, rtol=tol, atol=tol)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 703 TF_Tensor* result_tensor; in TF_TryEvaluateConstant_wrapper() local
705 TF_TryEvaluateConstant(graph, output, &result_tensor, status); in TF_TryEvaluateConstant_wrapper()
708 Safe_TF_TensorPtr safe_result_tensor(result_tensor); in TF_TryEvaluateConstant_wrapper()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_v1.py | 1869 result_tensor = value
1873 result_tensor = value
1874 metric_obj = result_tensor._metric_obj
1890 result_tensor = match(value)
1893 metric_obj, result_tensor = base_layer_utils.create_mean_metric(
|
| D | functional_test.py | 1829 result_tensor = network({
1833 result = self.evaluate(result_tensor)
1855 result_tensor = network(array_ops.ones((1, 1), 'float32'))
1856 result = self.evaluate(result_tensor)
1888 result_tensor = network(
1891 result = self.evaluate(result_tensor)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2399 Tensor result_tensor; in TF_TryEvaluateConstant() local
2402 graph->graph.versions().producer(), &evaluated, &result_tensor); in TF_TryEvaluateConstant()
2405 *result = TF_TensorFromTensor(result_tensor, &status->status); in TF_TryEvaluateConstant()
|