Searched refs:tensor_handles (Results 1 – 3 of 3) sorted by relevance
146 std::unique_ptr<TFE_TensorHandle*[]> tensor_handles( in Java_org_tensorflow_EagerOperationBuilder_addInputList() local149 tensor_handles[i] = requireTensorHandle(env, cinput_handles[i]); in Java_org_tensorflow_EagerOperationBuilder_addInputList()150 if (tensor_handles[i] == nullptr) { in Java_org_tensorflow_EagerOperationBuilder_addInputList()157 TFE_OpAddInputList(op, tensor_handles.get(), num_inputs, status); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
228 jlongArray tensor_handles) { in Java_org_tensorflow_GraphOperationBuilder_setAttrTensorList() argument231 const int n = env->GetArrayLength(tensor_handles); in Java_org_tensorflow_GraphOperationBuilder_setAttrTensorList()233 jlong* jhandles = env->GetLongArrayElements(tensor_handles, nullptr); in Java_org_tensorflow_GraphOperationBuilder_setAttrTensorList()239 env->ReleaseLongArrayElements(tensor_handles, jhandles, JNI_ABORT); in Java_org_tensorflow_GraphOperationBuilder_setAttrTensorList()
612 std::vector<tensorflow::TensorHandle*> tensor_handles; in TFE_CreatePackedTensorHandle() local613 tensor_handles.reserve(*num_handles); in TFE_CreatePackedTensorHandle()630 tensor_handles.push_back( in TFE_CreatePackedTensorHandle()637 std::move(tensor_handles), context, &handle); in TFE_CreatePackedTensorHandle()