| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 170 TensorHandle* packed_handle = nullptr; in TEST_F() local
172 &packed_handle)); in TEST_F()
179 EXPECT_EQ(packed_handle->NumPackedHandles(), 4); in TEST_F()
180 EXPECT_EQ(packed_handle->Type(), TensorHandle::PACKED); in TEST_F()
181 EXPECT_EQ(packed_handle->dtype, dtype); in TEST_F()
183 TF_ASSERT_OK(packed_handle->Shape(&packed_shape)); in TEST_F()
187 packed_handle->GetResourceHandleDtypesAndShapes(&dtypes_and_shapes)); in TEST_F()
193 reinterpret_cast<CompositeDevice*>(packed_handle->device()); in TEST_F()
200 for (int i = 0; i < packed_handle->NumPackedHandles(); ++i) { in TEST_F()
202 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(i, &h)); in TEST_F()
[all …]
|
| D | execute_node.cc | 38 TensorHandle* packed_handle) { in InitPackedHandle() argument
39 int num_handles = packed_handle->NumPackedHandles(); in InitPackedHandle()
44 TF_RETURN_IF_ERROR(packed_handle->ExtractPackedHandle(i, &h)); in InitPackedHandle()
|
| D | tensor_handle.h | 99 TensorHandle** packed_handle);
102 TensorHandle** packed_handle);
|
| D | tensor_handle.cc | 306 TensorHandle** packed_handle) { in CreatePackedHandle() argument
326 *packed_handle = in CreatePackedHandle()
328 (*packed_handle) in CreatePackedHandle()
335 TensorHandle** packed_handle) { in CreatePackedHandle() argument
346 /*device_name*/ "", ctx, packed_handle); in CreatePackedHandle()
|
| D | execute_node.h | 80 Device* input_device, TensorHandle* packed_handle);
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | variable.cc | 100 TensorHandle* packed_handle; in CreateUninitialized() local
102 std::move(handles), eager_ctx, &packed_handle)); in CreateUninitialized()
107 for (int i = 0; i != packed_handle->NumPackedHandles(); ++i) { in CreateUninitialized()
109 TF_RETURN_IF_ERROR(packed_handle->ExtractPackedHandle(i, &component)); in CreateUninitialized()
113 handle.reset(packed_handle); in CreateUninitialized()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_remote_test_util.cc | 128 TFE_TensorHandle* packed_handle = nullptr; in TestRemoteExecuteSilentCopies() local
132 packed_handle = TFE_CreatePackedTensorHandle(ctx, packed_handles.data(), in TestRemoteExecuteSilentCopies()
149 TFE_OpAddInput(matmul, has_packed_input ? packed_handle : h1_task2, status); in TestRemoteExecuteSilentCopies()
208 if (packed_handle) { in TestRemoteExecuteSilentCopies()
209 TFE_DeleteTensorHandle(packed_handle); in TestRemoteExecuteSilentCopies()
|
| D | c_api_distributed_test.cc | 175 TFE_TensorHandle* packed_handle = in TestFunctionWithPackedInput() local
178 EXPECT_EQ(TFE_TensorHandleDataType(packed_handle), TF_RESOURCE); in TestFunctionWithPackedInput()
179 EXPECT_EQ(TFE_TensorHandleNumDims(packed_handle, status), 0); in TestFunctionWithPackedInput()
180 EXPECT_EQ(TFE_TensorHandleNumElements(packed_handle, status), 1); in TestFunctionWithPackedInput()
184 EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status), in TestFunctionWithPackedInput()
186 EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status), in TestFunctionWithPackedInput()
198 TFE_OpAddInput(func, packed_handle, status); in TestFunctionWithPackedInput()
211 TFE_DeleteTensorHandle(packed_handle); in TestFunctionWithPackedInput()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 310 Status SerializePackedHandle(const uint64 op_id, TensorHandle* packed_handle, in SerializePackedHandle() argument
314 op->set_device_name(packed_handle->DeviceOrHostCPU(*ctx)->name()); in SerializePackedHandle()
315 for (int i = 0; i < packed_handle->NumPackedHandles(); ++i) { in SerializePackedHandle()
317 TF_RETURN_IF_ERROR(packed_handle->ExtractPackedHandle(i, &h)); in SerializePackedHandle()
|
| D | eager_service_impl_test.cc | 1163 tensorflow::TensorHandle* packed_handle; in TEST_F() local
1165 context_id, RemoteTensorHandleInternal(3, 0), &packed_handle)); in TEST_F()
1167 EXPECT_EQ(packed_handle->Type(), TensorHandle::PACKED); in TEST_F()
1168 EXPECT_EQ(packed_handle->NumPackedHandles(), 3); in TEST_F()
1169 EXPECT_EQ(packed_handle->device()->name(), composite_device); in TEST_F()
1172 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(0, &handle0)); in TEST_F()
1185 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(1, &handle1)); in TEST_F()
1193 TF_ASSERT_OK(packed_handle->ExtractPackedHandle(2, &handle2)); in TEST_F()
|
| D | eager_service_impl.cc | 731 tensorflow::TensorHandle* packed_handle = nullptr; in SendPackedHandle() local
736 send_packed_handle.device_name(), eager_context, &packed_handle)); in SendPackedHandle()
743 eager_context->RemoteMgr()->AddOperationOutputs({packed_handle}, in SendPackedHandle()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | packed_distributed_variable.py | 112 def packed_handle(self): member in PackedDistributedVariable
|
| D | values.py | 961 resource_map[self._packed_var.packed_handle] = resource_map[
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 235 TFE_TensorHandle* packed_handle = in InputTFE_OutputTensorHandles() local
239 EagerTensorFromHandle(packed_handle, /*is_packed=*/true); in InputTFE_OutputTensorHandles()
|