| /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 | 118 TF_Tensor* result_tensor; in TEST_P() local
119 s = GetValue(outputs[0], &result_tensor); in TEST_P()
121 auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
124 TF_DeleteTensor(result_tensor); in TEST_P()
125 result_tensor = nullptr; in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | get_options_op_test.cc | 91 Tensor result_tensor = output[0]; in TEST_F() local
92 string serialized_options = result_tensor.scalar<tstring>()(); in TEST_F()
95 TF_EXPECT_OK(ExpectEqual(expected_tensor, result_tensor)); in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| 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.cc | 133 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor) { in GetValue() argument
139 *result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in GetValue()
|
| D | unified_api_testutil.h | 90 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor);
|
| /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/core/kernels/mlir_generated/ |
| D | base_op.h | 132 Tensor result_tensor = ConvertDescriptorToTensor<OutputDataType>( in Compute() local
137 ctx, casted_result_tensor.BitcastFrom(result_tensor, CastedTfDataType, in Compute()
138 result_tensor.shape())); in Compute()
139 result_tensor = casted_result_tensor; in Compute()
142 ctx->set_output(0, result_tensor); in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | hashtable_ops_test.cc | 286 auto* result_tensor = interpreter_->tensor(kResultTensorId); in GetLookupResult() local
287 auto result_tensor_shape = GetTensorShape(result_tensor); in GetLookupResult()
288 auto* result_tensor_data = GetTensorData<ValueType>(result_tensor); in GetLookupResult()
299 auto* result_tensor = interpreter_->tensor(kResultTwoTensorId); in GetLookupTwoResult() local
300 auto result_tensor_shape = GetTensorShape(result_tensor); in GetLookupTwoResult()
301 auto* result_tensor_data = GetTensorData<ValueType>(result_tensor); in GetLookupTwoResult()
312 auto* result_tensor = interpreter_->tensor(kResultTensorId); in GetStringLookupResult() local
313 auto result_tensor_shape = GetTensorShape(result_tensor); in GetStringLookupResult()
318 auto string_ref = GetString(result_tensor, i); in GetStringLookupResult()
|
| /external/tensorflow/tensorflow/cc/experimental/libtf/tests/ |
| D | runtime_test.cc | 90 TF_ASSERT_OK_AND_ASSIGN(Tensor result_tensor, in TEST_P()
93 TF_ASSERT_OK(result_tensor.GetValue(absl::MakeSpan(out_val))); in TEST_P()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_op_handler.cc | 121 const RuntimeFallbackTensor& result_tensor, in GetDeviceFromFallbackTensor() argument
133 result_tensor.GetTensorHandle()->DeviceName(&status); in GetDeviceFromFallbackTensor()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| 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)
|
| D | sparse_ops_test.py | 855 def _check(self, result_tensor, result_np, input_sp_t): argument
856 self.assertTrue(isinstance(result_tensor, sparse_tensor.SparseTensor))
859 result_tensor.indices)
861 result_tensor.dense_shape)
864 result_tensor.indices, result_tensor.dense_shape, result_tensor.values)
|
| /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 | 1875 result_tensor = value
1879 result_tensor = value
1880 metric_obj = result_tensor._metric_obj
1896 result_tensor = match(value)
1899 metric_obj, result_tensor = base_layer_utils.create_mean_metric(
|
| D | functional_test.py | 1811 result_tensor = network({
1815 result = self.evaluate(result_tensor)
1837 result_tensor = network(array_ops.ones((1, 1), 'float32'))
1838 result = self.evaluate(result_tensor)
1870 result_tensor = network(
1873 result = self.evaluate(result_tensor)
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 785 auto result_tensor = MakeIndirectAsyncValue(host); in CreateLocalHandle() local
786 tensor_av.AndThen([host, result_tensor = result_tensor.CopyRef(), in CreateLocalHandle()
790 result_tensor->ForwardTo( in CreateLocalHandle()
795 result_tensor->ForwardTo(tensor_av.CopyRef()); in CreateLocalHandle()
800 AsyncValueRef<Tensor>(std::move(result_tensor)))), in CreateLocalHandle()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 2420 Tensor result_tensor; in TF_TryEvaluateConstant() local
2423 graph->graph.versions().producer(), &evaluated, &result_tensor); in TF_TryEvaluateConstant()
2426 *result = TF_TensorFromTensor(result_tensor, &status->status); in TF_TryEvaluateConstant()
|